public class TimeIterationListener extends BaseTrainingListener implements Serializable
| Constructor and Description |
|---|
TimeIterationListener(int iterationCount)
Constructor
|
TimeIterationListener(int frequency,
int iterationCount)
Constructor
|
| Modifier and Type | Method and Description |
|---|---|
void |
iterationDone(Model model,
int iteration,
int epoch)
Event listener for each iteration.
|
onBackwardPass, onEpochEnd, onEpochStart, onForwardPass, onForwardPass, onGradientCalculationpublic TimeIterationListener(int iterationCount)
iterationCount - The global number of iteration for training (all epochs)public TimeIterationListener(int frequency,
int iterationCount)
frequency - frequency with which to print the remaining timeiterationCount - The global number of iteration for training (all epochs)public void iterationDone(Model model, int iteration, int epoch)
TrainingListeneriterationDone in interface TrainingListeneriterationDone in class BaseTrainingListenermodel - the model iteratingiteration - the iterationCopyright © 2021. All rights reserved.