Package ai.djl.training.optimizer
Class RmsProp.Builder
- java.lang.Object
-
- ai.djl.training.optimizer.Optimizer.OptimizerBuilder<RmsProp.Builder>
-
- ai.djl.training.optimizer.RmsProp.Builder
-
- Enclosing class:
- RmsProp
public static final class RmsProp.Builder extends Optimizer.OptimizerBuilder<RmsProp.Builder>
The Builder to construct anRmsProp
object.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description RmsProp
build()
Builds aRmsProp
block.RmsProp.Builder
optCentered(boolean centered)
Sets which version of RMSProp to use.RmsProp.Builder
optEpsilon(float epsilon)
Sets \(epsilon\) - a small quantity for numerical stability.RmsProp.Builder
optLearningRateTracker(Tracker learningRateTracker)
Sets theTracker
for this optimizer.RmsProp.Builder
optMomentum(float momentum)
Sets the momentum factor.RmsProp.Builder
optRho(float rho)
Sets the decay factor for the moving average over the past squared gradient.protected RmsProp.Builder
self()
-
Methods inherited from class ai.djl.training.optimizer.Optimizer.OptimizerBuilder
optBeginNumUpdate, optClipGrad, optWeightDecays, setRescaleGrad
-
-
-
-
Method Detail
-
self
protected RmsProp.Builder self()
- Specified by:
self
in classOptimizer.OptimizerBuilder<RmsProp.Builder>
-
optLearningRateTracker
public RmsProp.Builder optLearningRateTracker(Tracker learningRateTracker)
Sets theTracker
for this optimizer.- Parameters:
learningRateTracker
- theTracker
to be set- Returns:
- this
Builder
-
optRho
public RmsProp.Builder optRho(float rho)
Sets the decay factor for the moving average over the past squared gradient.- Parameters:
rho
- the decay factor for the moving average over past squared gradient- Returns:
- this
Builder
-
optMomentum
public RmsProp.Builder optMomentum(float momentum)
Sets the momentum factor. This is only used if centered is set to true.- Parameters:
momentum
- the momentum factor- Returns:
- this
Builder
-
optEpsilon
public RmsProp.Builder optEpsilon(float epsilon)
Sets \(epsilon\) - a small quantity for numerical stability.- Parameters:
epsilon
- a small quantity for numerical stability- Returns:
- this
Builder
-
optCentered
public RmsProp.Builder optCentered(boolean centered)
Sets which version of RMSProp to use.True: Grave's version False: Tieleman and Hinton's version
- Parameters:
centered
- the RMSProp version- Returns:
- this
Builder
-
-