Package ai.djl.training.listener

Contains classes providing functionality during training through TrainingListener.

It contains a main interface TrainingListener and various listeners that extend it.

Interface Summary

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 Summary

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.