Package ai.djl.training
Contains classes and implementations to train a neural network.
- See Also:
Trainer
-
Interface Summary Interface Description GradientCollector An interface that provides a mechanism to collect gradients during training.ParameterServer An interface for a key-value store to store parameters, and their corresponding gradients.TrainingConfig An interface that is responsible for holding the configuration required byTrainer
. -
Class Summary Class Description DefaultTrainingConfig DefaultTrainingConfig
is an implementation of theTrainingConfig
interface.EasyTrain Helper for easy training of a whole model, a trainining batch, or a validation batch.LocalParameterServer LocalParameterServer
is an implementation of theParameterServer
interface.ParameterStore TheParameterStore
contains a map from a parameter to the mirrors of it on other devices.Trainer TheTrainer
interface provides a session for model training.TrainingResult A class that is responsible for holding the training result produced byTrainer
.