MPSNNOptimizerStochasticGradientDescent
An optimization layer that performs a gradient descent with an optional momentum update.
Declaration
class MPSNNOptimizerStochasticGradientDescentTopics
Initializers
init(device:learningRate:)init(device:momentumScale:useNesterovMomentum:optimizerDescriptor:)init(device:momentumScale:useNestrovMomentum:optimizerDescriptor:)
Instance Properties
Instance Methods
encode(commandBuffer:batchNormalizationGradientState:batchNormalizationSourceState:inputMomentumVectors:resultState:)encode(commandBuffer:batchNormalizationState:inputMomentumVectors:resultState:)encode(commandBuffer:convolutionGradientState:convolutionSourceState:inputMomentumVectors:resultState:)encode(commandBuffer:inputGradientMatrix:inputValuesMatrix:inputMomentumMatrix:resultValuesMatrix:)encode(commandBuffer:inputGradientVector:inputValuesVector:inputMomentumVector:resultValuesVector:)