ai.djl.training (Deep Java Library 0.3.0 API specification)

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	The `Trainer` interface provides a session for model training.
TrainingConfig	An interface that is responsible for holding the configuration required by `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.

Package ai.djl.training Description

Contains classes and implementations to train a neural network.