org.nd4j.autodiff.validation.listeners

Class NonInplaceValidationListener

java.lang.Object

org.nd4j.autodiff.listeners.BaseListener

org.nd4j.autodiff.validation.listeners.NonInplaceValidationListener

All Implemented Interfaces:

Listener

public class NonInplaceValidationListener
extends BaseListener

Field Summary

Fields

Modifier and Type	Field and Description
protected INDArray[]	opInputs

Constructor Summary

Constructors

Constructor and Description
NonInplaceValidationListener()

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, INDArray[] outputs) Called at the end of each operation execution
void	preOpExecution(SameDiff sd, At at, SameDiffOp op) Called just before each operation is executed (native code called, etc) - after all inputs etc have been set

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

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

Methods inherited from class java.lang.Object

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

Field Detail

opInputs

protected INDArray[] opInputs

Constructor Detail

NonInplaceValidationListener

public NonInplaceValidationListener()

Method Detail

preOpExecution

public void preOpExecution(SameDiff sd,
                           At at,
                           SameDiffOp op)

Description copied from interface: Listener

Called just before each operation is executed (native code called, etc) - after all inputs etc have been set

Specified by:

preOpExecution in interface Listener

Overrides:

preOpExecution in class BaseListener

Parameters:

sd - The SameDiff instance

at - Current iteration/epoch etc

op - Operation that has just been executed

opExecution

public void opExecution(SameDiff sd,
                        At at,
                        MultiDataSet batch,
                        SameDiffOp op,
                        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

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.