Package ai.djl.training.optimizer

Contains classes for optimizing a neural network Block.

See: Description

Class Summary

Class	Description
Adam	Adam is a generalization of the AdaGrad Optimizer.
Adam.Builder	The Builder to construct an Adam object.
Nag	Nag is a Nesterov accelerated gradient optimizer.
Nag.Builder	The Builder to construct an Nag object.
Optimizer	An Optimizer updates the weight parameters to minimize the loss function.
Optimizer.OptimizerBuilder<T extends Optimizer.OptimizerBuilder>	The Builder to construct an Optimizer.
Sgd	Sgd is a Stochastic Gradient Descent (SDG) optimizer.
Sgd.Builder	The Builder to construct an Sgd object.

Package ai.djl.training.optimizer Description

Contains classes for optimizing a neural network Block.

It contains a main interface Optimizer and various optimizers that extend it. There are also the helpers for learning rates in ai.djl.training.optimizer.learningrate.