TrainingListener
.See: Description
Interface | Description |
---|---|
TrainingListener |
TrainingListener offers an interface that performs some actions when certain events have
occurred in the Trainer . |
TrainingListener.Defaults |
Contains default
TrainingListener sets. |
Class | Description |
---|---|
DivergenceCheckTrainingListener |
TrainingListener that gives early warning if your training has failed by divergence. |
EpochTrainingListener |
EpochTrainingListener that tracks epochs. |
EvaluatorTrainingListener |
TrainingListener that records evaluator results. |
LoggingTrainingListener |
TrainingListener that outputs the progress of training each batch and epoch into logs. |
MemoryTrainingListener |
TrainingListener that collects the memory usage information. |
SaveModelTrainingListener |
A
TrainingListener that saves a model and can save checkpoints. |
TimeMeasureTrainingListener |
TrainingListener that outputs the training time metrics after training is done. |
TrainingListener.BatchData |
A class to pass data from the batch into the training listeners.
|
TrainingListenerAdapter |
Base implementation of the training listener that does nothing.
|
TrainingListener
.
It contains a main interface TrainingListener
and various
listeners that extend it.