Package ai.djl.training

package ai.djl.training

Contains classes and implementations to train a neural network.

See Also:

• Trainer

Related Packages

Package	Description
ai.djl	Contains top level, engine-agnostic classes for both inference and training.
ai.djl.training.dataset	Contains classes to download and prepare training and testing data.
ai.djl.training.evaluator	Contains classes for evaluating the effectiveness of models.
ai.djl.training.hyperparameter	Contains utilities to train, describe, and manipulate Hyperparameters.
ai.djl.training.initializer	Contains classes for initializing a Block's Parameters.
ai.djl.training.listener	Contains classes providing functionality during training through TrainingListener.
ai.djl.training.loss	Contains classes for measuring the Loss of a model.
ai.djl.training.optimizer	Contains classes for optimizing a neural network Block.
ai.djl.training.tracker	Contains classes for having a gradually changing hyper-parameter.
ai.djl.training.util	Contains utilities to use during training.

Class	Description
DefaultTrainingConfig	DefaultTrainingConfig is an implementation of the TrainingConfig interface.
EasyTrain	Helper for easy training of a whole model, a trainining batch, or a validation batch.
GradientCollector	An interface that provides a mechanism to collect gradients during training.
LocalParameterServer	LocalParameterServer is an implementation of the ParameterServer interface.
ParameterServer	An interface for a key-value store to store parameters, and their corresponding gradients.
ParameterStore	The ParameterStore contains a map from a parameter to the mirrors of it on other devices.
Trainer	The Trainer interface provides a session for model training.
TrainingConfig	An interface that is responsible for holding the configuration required by Trainer.
TrainingResult	A class that is responsible for holding the training result produced by Trainer.