public class NonInplaceValidationListener extends BaseListener
Modifier and Type | Field and Description |
---|---|
protected INDArray[] |
opInputs |
protected INDArray[] |
opInputsOrig |
Constructor and Description |
---|
NonInplaceValidationListener() |
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
|
void |
preOpExecution(SameDiff sd,
At at,
SameDiffOp op,
OpContext oc)
Called just before each operation is executed (native code called, etc) - after all inputs etc have been set
|
activationAvailable, epochEnd, epochStart, iterationDone, iterationStart, operationEnd, operationStart, preUpdate, requiredVariables, validationDone
public void preOpExecution(SameDiff sd, At at, SameDiffOp op, OpContext oc)
Listener
preOpExecution
in interface Listener
preOpExecution
in class BaseListener
sd
- The SameDiff instanceat
- Current iteration/epoch etcop
- Operation that has just been executedpublic 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 © 2020. All rights reserved.