Package ai.djl.training.listener
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.
-
ClassDescription
TrainingListenerthat gives early warning if your training has failed by divergence.Listener that allows the training to be stopped early if the validation loss is not improving, or if time has expired.A builder for aEarlyStoppingListener.Thrown when training is stopped early, the message will contain the reason why it is stopped early.EpochTrainingListenerthat tracks epochs.TrainingListenerthat records evaluator results.TrainingListenerthat outputs the progress of training each batch and epoch into logs.TrainingListenerthat collects the memory usage information.ATrainingListenerthat saves a model and can save checkpoints.TrainingListenerthat outputs the training time metrics after training is done.TrainingListeneroffers an interface that performs some actions when certain events have occurred in theTrainer.A class to pass data from the batch into the training listeners.Contains defaultTrainingListenersets.Base implementation of the training listener that does nothing.