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.

Beta Software

This documentation contains preliminary information about an API or technology in development. This information is subject to change, and software implemented according to this documentation should be tested with final operating system software.

Learn more about using Apple's beta software