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(Tracker learningRateTracker)
Sets the
Tracker for this optimizer. |
optBeginNumUpdate, optClipGrad, optWeightDecays, setRescaleGrad
protected Sgd.Builder self()
self
in class Optimizer.OptimizerBuilder<Sgd.Builder>
public Sgd.Builder setLearningRateTracker(Tracker learningRateTracker)
Tracker
for this optimizer.learningRateTracker
- the Tracker
to be setBuilder
public Sgd.Builder optMomentum(float momentum)
Sgd
.momentum
- the value of momentumBuilder