public static final class Sgd.Builder extends Optimizer.OptimizerBuilder<Sgd.Builder>
Sgd
object.Modifier and Type | Method and Description |
---|---|
Sgd |
build()
Builds a
Sgd block. |
Sgd.Builder |
optMomentum(float momentum)
Sets the momentum for
Sgd . |
protected Sgd.Builder |
self() |
Sgd.Builder |
setLearningRateTracker(LearningRateTracker learningRateTracker)
Sets the
LearningRateTracker for this optimizer. |
optBeginNumUpdate, optClipGrad, optWeightDecays, setRescaleGrad
protected Sgd.Builder self()
self
in class Optimizer.OptimizerBuilder<Sgd.Builder>
public Sgd.Builder setLearningRateTracker(LearningRateTracker learningRateTracker)
LearningRateTracker
for this optimizer.learningRateTracker
- the LearningRateTracker
to be setBuilder
public Sgd.Builder optMomentum(float momentum)
Sgd
.momentum
- the value of momentumBuilder