init(descriptor:momentumScale:usesNesterovMomentum:)
Create an SGD optimizer with the descriptor, momentum scale, and option to enable Nesterov momentum that you specify.
Declaration
convenience init(descriptor optimizerDescriptor: MLCOptimizerDescriptor, momentumScale: Float, usesNesterovMomentum: Bool)Parameters
- optimizerDescriptor:
An object you use to configure the optimizer.
- momentumScale:
The momentum scale.
- usesNesterovMomentum:
A Boolean that indicates whether you enable Nesterov momentum.