Package ai.djl.training

Contains classes and implementations to train a neural network.

See: Description

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.
Trainer	An interface that represents a single training iteration.
TrainingConfig	An interface that is responsible for holding the configuration required by Trainer.
TrainingListener	TrainingListener offers an interface that allows performing some actions when certain events have occurred in the Trainer.

Class Summary

Class	Description
DefaultTrainingConfig	DefaultTrainingConfig is an implementation of the TrainingConfig interface.
LocalParameterServer	LocalParameterServer is an implementation of the ParameterServer interface.
ParameterStore	The ParameterStore contains a map from a parameter to the mirrors of it on other devices.

Enum Summary

Enum	Description
GradReq	An enum that indicates whether gradient is required.

Package ai.djl.training Description

Contains classes and implementations to train a neural network.

See Also:

Trainer