org.nd4j.autodiff.validation

Class ActivationGradientCheckListener

java.lang.Object

org.nd4j.autodiff.listeners.BaseListener

org.nd4j.autodiff.validation.ActivationGradientCheckListener

All Implemented Interfaces:

Listener

public class ActivationGradientCheckListener
extends BaseListener

A listener used for debugging and testing purposes - specifically for gradient checking activations internally in GradCheckUtil. It probably isn't useful for anything outside of this.

Author:

Alex Black

Constructor Summary

Constructors

Constructor and Description
ActivationGradientCheckListener()

Method Summary

Modifier and Type	Method and Description
boolean	isActive(Operation operation) Returns whether this listener is active during the given operation.
void	opExecution(SameDiff sd, At at, MultiDataSet batch, SameDiffOp op, OpContext opContext, INDArray[] outputs) Called at the end of each operation execution

Methods inherited from class org.nd4j.autodiff.listeners.BaseListener

activationAvailable, epochEnd, epochStart, iterationDone, iterationStart, operationEnd, operationStart, preOpExecution, preUpdate, requiredVariables, validationDone

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

ActivationGradientCheckListener

public ActivationGradientCheckListener()

Method Detail

isActive

public boolean isActive(Operation operation)

Description copied from interface: Listener

Returns whether this listener is active during the given operation. If this returns false for the given operation, those listener methods will not be called.

opExecution

public void opExecution(SameDiff sd,
                        At at,
                        MultiDataSet batch,
                        SameDiffOp op,
                        OpContext opContext,
                        INDArray[] outputs)

Description copied from interface: Listener

Called at the end of each operation execution

Note: Outputs will most likely be freed later, use detach() if you need to save it.

Specified by:

opExecution in interface Listener

Overrides:

opExecution in class BaseListener

Parameters:

sd - The SameDiff instance

at - Current iteration/epoch etc

batch - The batch's input data. May be null if not called with a batch

op - Operation that has just been executed

outputs - The output arrays for the just-executed operation