Package ai.djl.training.optimizer
Class Adagrad.Builder
- java.lang.Object
-
- ai.djl.training.optimizer.Optimizer.OptimizerBuilder<Adagrad.Builder>
-
- ai.djl.training.optimizer.Adagrad.Builder
-
- Enclosing class:
- Adagrad
public static final class Adagrad.Builder extends Optimizer.OptimizerBuilder<Adagrad.Builder>
The Builder to construct anAdagradobject.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description Adagradbuild()Builds aAdagradblock.Adagrad.BuilderoptEpsilon(float epsilon)Sets \(epsilon\) - a small quantity for numerical stability.Adagrad.BuilderoptLearningRateTracker(ParameterTracker learningRateTracker)Sets theParameterTrackerfor this optimizer.protected Adagrad.Builderself()-
Methods inherited from class ai.djl.training.optimizer.Optimizer.OptimizerBuilder
optBeginNumUpdate, optClipGrad, optWeightDecays, setRescaleGrad
-
-
-
-
Method Detail
-
self
protected Adagrad.Builder self()
- Specified by:
selfin classOptimizer.OptimizerBuilder<Adagrad.Builder>
-
optLearningRateTracker
public Adagrad.Builder optLearningRateTracker(ParameterTracker learningRateTracker)
Sets theParameterTrackerfor this optimizer.- Parameters:
learningRateTracker- theParameterTrackerto be set- Returns:
- this
Builder
-
optEpsilon
public Adagrad.Builder optEpsilon(float epsilon)
Sets \(epsilon\) - a small quantity for numerical stability.- Parameters:
epsilon- a small quantity for numerical stability- Returns:
- this
Builder
-
-