Class

MPSNNOptimizerStochasticGradientDescent

An optimization layer that performs a gradient descent with an optional momentum update.

Declaration

@interface MPSNNOptimizerStochasticGradientDescent : MPSNNOptimizer

Relationships

Inherits From

See Also

Optimization Layers

MPSNNOptimizerAdam

An optimization layer that performs an Adam pdate.

MPSNNOptimizerRMSProp

An optimization layer that performs a root mean square propagation update.

MPSNNOptimizer

The base class for optimization layers.

MPSNNOptimizerDescriptor

An object that specifies properties used by an optimizer kernel.