See: Description
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 | 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. |
Trainer