Package 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
-
-
Constructor Summary
Constructors Constructor Description ActivationGradientCheckListener()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method 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
-
-
-
-
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 interfaceListener
- Overrides:
opExecution
in classBaseListener
- Parameters:
sd
- The SameDiff instanceat
- Current iteration/epoch etcbatch
- The batch's input data. May be null if not called with a batchop
- Operation that has just been executedoutputs
- The output arrays for the just-executed operation
-
-