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