public class FailureTestingListener extends Object implements TrainingListener, Serializable
| Modifier and Type | Class and Description |
|---|---|
static class |
FailureTestingListener.And |
static class |
FailureTestingListener.CallType |
static class |
FailureTestingListener.FailureMode |
static class |
FailureTestingListener.FailureTrigger |
static class |
FailureTestingListener.HostNameTrigger |
static class |
FailureTestingListener.IterationEpochTrigger |
static class |
FailureTestingListener.Or |
static class |
FailureTestingListener.RandomProb |
static class |
FailureTestingListener.TimeSinceInitializedTrigger |
static class |
FailureTestingListener.UserNameTrigger |
| Constructor and Description |
|---|
FailureTestingListener(@NonNull FailureTestingListener.FailureMode mode,
@NonNull FailureTestingListener.FailureTrigger trigger) |
| Modifier and Type | Method and Description |
|---|---|
protected void |
call(FailureTestingListener.CallType callType,
Model model) |
void |
iterationDone(Model model,
int iteration,
int epoch)
Event listener for each iteration.
|
void |
onBackwardPass(Model model)
Called once per iteration (backward pass) after gradients have been calculated, and updated
Gradients are available via
Model.gradient(). |
void |
onEpochEnd(Model model)
Called once at the end of each epoch, when using methods such as
MultiLayerNetwork.fit(DataSetIterator),
ComputationGraph.fit(DataSetIterator) or ComputationGraph.fit(MultiDataSetIterator) |
void |
onEpochStart(Model model)
Called once at the start of each epoch, when using methods such as
MultiLayerNetwork.fit(DataSetIterator),
ComputationGraph.fit(DataSetIterator) or ComputationGraph.fit(MultiDataSetIterator) |
void |
onForwardPass(Model model,
List<INDArray> activations)
Called once per iteration (forward pass) for activations (usually for a
MultiLayerNetwork),
only at training time |
void |
onForwardPass(Model model,
Map<String,INDArray> activations)
Called once per iteration (forward pass) for activations (usually for a
ComputationGraph),
only at training time |
void |
onGradientCalculation(Model model)
Called once per iteration (backward pass) before the gradients are updated
Gradients are available via
Model.gradient(). |
public FailureTestingListener(@NonNull
@NonNull FailureTestingListener.FailureMode mode,
@NonNull
@NonNull FailureTestingListener.FailureTrigger trigger)
public void iterationDone(Model model, int iteration, int epoch)
TrainingListeneriterationDone in interface TrainingListenermodel - the model iteratingiteration - the iterationpublic void onEpochStart(Model model)
TrainingListenerMultiLayerNetwork.fit(DataSetIterator),
ComputationGraph.fit(DataSetIterator) or ComputationGraph.fit(MultiDataSetIterator)onEpochStart in interface TrainingListenerpublic void onEpochEnd(Model model)
TrainingListenerMultiLayerNetwork.fit(DataSetIterator),
ComputationGraph.fit(DataSetIterator) or ComputationGraph.fit(MultiDataSetIterator)onEpochEnd in interface TrainingListenerpublic void onForwardPass(Model model, List<INDArray> activations)
TrainingListenerMultiLayerNetwork),
only at training timeonForwardPass in interface TrainingListenermodel - Modelactivations - Layer activations (including input)public void onForwardPass(Model model, Map<String,INDArray> activations)
TrainingListenerComputationGraph),
only at training timeonForwardPass in interface TrainingListenermodel - Modelactivations - Layer activations (including input)public void onGradientCalculation(Model model)
TrainingListenerModel.gradient().
Note that gradients will likely be updated in-place - thus they should be copied or processed synchronously
in this method.
For updates (gradients post learning rate/momentum/rmsprop etc) see TrainingListener.onBackwardPass(Model)
onGradientCalculation in interface TrainingListenermodel - Modelpublic void onBackwardPass(Model model)
TrainingListenerModel.gradient().
Unlike TrainingListener.onGradientCalculation(Model) the gradients at this point will be post-update, rather than
raw (pre-update) gradients at that method call.
onBackwardPass in interface TrainingListenermodel - Modelprotected void call(FailureTestingListener.CallType callType, Model model)
Copyright © 2021. All rights reserved.