Momentum

class Momentum(learningRate: Float, momentum: Float, useNesterov: Boolean, clipGradient: ClipGradientAction) : Optimizer
Content copied to clipboard

Improved version of SGD optimizer.

Constructors

Momentum
Link copied to clipboard
fun Momentum(learningRate: Float = 0.001f, momentum: Float = 0.99f, useNesterov: Boolean = true, clipGradient: ClipGradientAction = NoClipGradient())
Content copied to clipboard

Properties

clipGradient
Link copied to clipboard
val clipGradient: ClipGradientAction
Content copied to clipboard

Strategy of gradient clipping as subclass of ClipGradientAction.

optimizerName
Link copied to clipboard
open override val optimizerName: String
Content copied to clipboard

Returns optimizer name.