Modifier and Type | Method and Description |
---|---|
At |
At.copy() |
At |
At.copy(Operation operation) |
static At |
At.defaultAt() |
static At |
At.defaultAt(Operation op) |
Modifier and Type | Method and Description |
---|---|
void |
BaseListener.activationAvailable(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
String varName,
INDArray activation) |
void |
Listener.activationAvailable(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
String varName,
INDArray activation)
Called when any activation becomes available.
|
void |
BaseEvaluationListener.activationAvailable(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
String varName,
INDArray activation) |
void |
BaseEvaluationListener.activationAvailableEvaluations(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
String varName,
INDArray activation)
|
ListenerResponse |
BaseListener.epochEnd(SameDiff sd,
At at,
LossCurve lossCurve,
long epochTimeMillis) |
ListenerResponse |
Listener.epochEnd(SameDiff sd,
At at,
LossCurve lossCurve,
long epochTimeMillis)
Called at the end of every epoch, when fitting from an iterator
|
ListenerResponse |
BaseEvaluationListener.epochEnd(SameDiff sd,
At at,
LossCurve lossCurve,
long epochTimeMillis) |
ListenerResponse |
BaseEvaluationListener.epochEndEvaluations(SameDiff sd,
At at,
LossCurve lossCurve,
long epochTimeMillis,
EvaluationRecord evaluations)
See
Listener.epochEnd(SameDiff, At, LossCurve, long) , also provided the requested evaluations |
void |
BaseListener.epochStart(SameDiff sd,
At at) |
void |
Listener.epochStart(SameDiff sd,
At at)
Called at the start of every epoch, when fitting from an iterator
|
void |
BaseEvaluationListener.epochStart(SameDiff sd,
At at) |
void |
BaseEvaluationListener.epochStartEvaluations(SameDiff sd,
At at)
|
void |
BaseListener.iterationDone(SameDiff sd,
At at,
MultiDataSet dataSet,
Loss loss) |
void |
Listener.iterationDone(SameDiff sd,
At at,
MultiDataSet dataSet,
Loss loss)
Called at the end of every iteration, after all operations (including updating parameters) has been completed
|
void |
BaseListener.iterationStart(SameDiff sd,
At at,
MultiDataSet data,
long etlMs) |
void |
Listener.iterationStart(SameDiff sd,
At at,
MultiDataSet data,
long etlTimeMs)
Called at the start of every iteration (minibatch), before any operations have been executed
|
void |
BaseListener.opExecution(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
INDArray[] outputs) |
void |
Listener.opExecution(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
INDArray[] outputs)
Called at the end of each operation execution
|
void |
BaseListener.preOpExecution(SameDiff sd,
At at,
SameDiffOp op) |
void |
Listener.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
|
void |
BaseListener.preUpdate(SameDiff sd,
At at,
Variable v,
INDArray update) |
void |
Listener.preUpdate(SameDiff sd,
At at,
Variable v,
INDArray update)
Called just before each parameter is to be updated - i.e., just before each parameter is modified.
|
ListenerResponse |
BaseListener.validationDone(SameDiff sd,
At at,
long validationTimeMillis) |
ListenerResponse |
Listener.validationDone(SameDiff sd,
At at,
long validationTimeMillis)
Called after the end of every epoch, once validation evaluation is done, when training
|
ListenerResponse |
BaseEvaluationListener.validationDone(SameDiff sd,
At at,
long validationTimeMillis) |
ListenerResponse |
BaseEvaluationListener.validationDoneEvaluations(SameDiff sd,
At at,
long validationTimeMillis,
EvaluationRecord evaluations)
See
Listener.validationDone(SameDiff, At, long) , also provided the requested evaluations |
Modifier and Type | Method and Description |
---|---|
ListenerResponse |
CheckpointListener.epochEnd(SameDiff sameDiff,
At at,
LossCurve lossCurve,
long epochTimeMillis) |
void |
CheckpointListener.iterationDone(SameDiff sd,
At at,
MultiDataSet dataSet,
Loss loss) |
Modifier and Type | Method and Description |
---|---|
void |
OpBenchmarkListener.opExecution(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
INDArray[] outputs) |
void |
ArraySavingListener.opExecution(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
INDArray[] outputs) |
void |
OpBenchmarkListener.preOpExecution(SameDiff sd,
At at,
SameDiffOp op) |
void |
ExecDebuggingListener.preOpExecution(SameDiff sd,
At at,
SameDiffOp op) |
Modifier and Type | Method and Description |
---|---|
ListenerResponse |
UIListener.epochEnd(SameDiff sd,
At at,
LossCurve lossCurve,
long epochTimeMillis) |
ListenerResponse |
ScoreListener.epochEnd(SameDiff sd,
At at,
LossCurve lossCurve,
long epochTimeMillis) |
ListenerResponse |
HistoryListener.epochEndEvaluations(SameDiff sd,
At at,
LossCurve lossCurve,
long epochTimeMillis,
EvaluationRecord evaluations) |
void |
UIListener.epochStart(SameDiff sd,
At at) |
void |
ScoreListener.epochStart(SameDiff sd,
At at) |
void |
UIListener.iterationDone(SameDiff sd,
At at,
MultiDataSet dataSet,
Loss loss) |
void |
ScoreListener.iterationDone(SameDiff sd,
At at,
MultiDataSet dataSet,
Loss loss) |
void |
UIListener.iterationStart(SameDiff sd,
At at,
MultiDataSet data,
long etlMs) |
void |
ScoreListener.iterationStart(SameDiff sd,
At at,
MultiDataSet data,
long etlMs) |
void |
UIListener.opExecution(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
INDArray[] outputs) |
void |
UIListener.preUpdate(SameDiff sd,
At at,
Variable v,
INDArray update) |
ListenerResponse |
HistoryListener.validationDoneEvaluations(SameDiff sd,
At at,
long validationTimeMillis,
EvaluationRecord evaluations) |
Modifier and Type | Method and Description |
---|---|
protected Map<String,INDArray> |
SameDiff.directExecHelper(Map<String,INDArray> placeholders,
At at,
MultiDataSet batch,
Collection<String> requiredActivations,
List<Listener> activeListeners,
String... outputs)
Do inference for the given variables for a single batch, with training information
|
Modifier and Type | Method and Description |
---|---|
abstract T[] |
AbstractSession.getOutputs(O op,
AbstractSession.FrameIter outputFrameIter,
Set<AbstractSession.VarId> inputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
List<Listener> listeners,
At at,
MultiDataSet batch,
Set<String> allReqVariables)
Execute the op - calculate INDArrays, or shape info, etc
|
INDArray[] |
TrainingSession.getOutputs(SameDiffOp op,
AbstractSession.FrameIter outputFrameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
List<Listener> listeners,
At at,
MultiDataSet batch,
Set<String> allReqVariables) |
INDArray[] |
InferenceSession.getOutputs(SameDiffOp op,
AbstractSession.FrameIter outputFrameIter,
Set<AbstractSession.VarId> opInputs,
Set<AbstractSession.VarId> allIterInputs,
Set<String> constAndPhInputs,
List<Listener> listeners,
At at,
MultiDataSet batch,
Set<String> allReqVariables) |
Map<String,T> |
AbstractSession.output(List<String> variables,
Map<String,T> placeholderValues,
MultiDataSet batch,
Collection<String> requiredActivations,
List<Listener> listeners,
At at)
Get the output of the session - i.e., perform inference/forward pass and return the autputs for the specified variables
|
protected Map<String,INDArray> |
InferenceSession.preprocessPlaceholders(Map<String,INDArray> placeholders,
At at) |
protected Map<String,T> |
AbstractSession.preprocessPlaceholders(Map<String,T> placeholders,
At at)
Preprocess the placeholder values, if required.
|
Loss |
TrainingSession.trainingIteration(TrainingConfig config,
Map<String,INDArray> placeholders,
Set<String> paramsToTrain,
Map<String,GradientUpdater> updaters,
MultiDataSet batch,
List<String> lossVariables,
List<Listener> listeners,
At at)
Perform one iteration of training - i.e., do forward and backward passes, and update the parameters
|
Modifier and Type | Method and Description |
---|---|
void |
ActivationGradientCheckListener.opExecution(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
INDArray[] outputs) |
Modifier and Type | Method and Description |
---|---|
void |
NonInplaceValidationListener.opExecution(SameDiff sd,
At at,
MultiDataSet batch,
SameDiffOp op,
INDArray[] outputs) |
void |
NonInplaceValidationListener.preOpExecution(SameDiff sd,
At at,
SameDiffOp op) |
Copyright © 2019. All rights reserved.