Package ai.djl.training.optimizer
Class Nag.Builder
- java.lang.Object
-
- ai.djl.training.optimizer.Optimizer.OptimizerBuilder<Nag.Builder>
-
- ai.djl.training.optimizer.Nag.Builder
-
- Enclosing class:
- Nag
public static final class Nag.Builder extends Optimizer.OptimizerBuilder<Nag.Builder>
The Builder to construct anNagobject.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description Nagbuild()Builds aNagblock.protected Nag.Builderself()Nag.BuildersetLearningRateTracker(ParameterTracker learningRateTracker)Sets theParameterTrackerfor this optimizer.Nag.BuildersetMomentum(float momentum)Sets the momentum forNag.-
Methods inherited from class ai.djl.training.optimizer.Optimizer.OptimizerBuilder
optBeginNumUpdate, optClipGrad, optWeightDecays, setRescaleGrad
-
-
-
-
Method Detail
-
setLearningRateTracker
public Nag.Builder setLearningRateTracker(ParameterTracker learningRateTracker)
Sets theParameterTrackerfor this optimizer.- Parameters:
learningRateTracker- theParameterTrackerto be set- Returns:
- this
Builder
-
setMomentum
public Nag.Builder setMomentum(float momentum)
Sets the momentum forNag.- Parameters:
momentum- the value of momentum- Returns:
- this
Builder
-
self
protected Nag.Builder self()
- Specified by:
selfin classOptimizer.OptimizerBuilder<Nag.Builder>
-
-