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