public class ActivationGradientCheckListener extends BaseListener
Constructor and Description |
---|
ActivationGradientCheckListener() |
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
|
activationAvailable, epochEnd, epochStart, iterationDone, iterationStart, operationEnd, operationStart, preOpExecution, preUpdate, requiredVariables, validationDone
public boolean isActive(Operation operation)
Listener
public void opExecution(SameDiff sd, At at, MultiDataSet batch, SameDiffOp op, OpContext opContext, INDArray[] outputs)
Listener
Note: Outputs will most likely be freed later, use detach() if you need to save it.
opExecution
in interface Listener
opExecution
in class BaseListener
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 operationCopyright © 2021. All rights reserved.