Package ai.djl.training.tracker
Class PolynomialDecayTracker.Builder
- java.lang.Object
-
- ai.djl.training.tracker.PolynomialDecayTracker.Builder
-
- Enclosing class:
- PolynomialDecayTracker
public static final class PolynomialDecayTracker.Builder extends java.lang.ObjectBuilder for PolynomialDecayTracker.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description PolynomialDecayTrackerbuild()Builds a PolynomialDecayTracker.PolynomialDecayTracker.BuilderoptPower(float power)Sets the power of the polynomial to decay by.PolynomialDecayTracker.BuildersetBaseValue(float baseValue)Sets the initial value after no steps.PolynomialDecayTracker.BuildersetDecaySteps(int decaySteps)Sets the number of training steps to decay learning rate in.PolynomialDecayTracker.BuildersetEndLearningRate(float endLearningRate)Sets the learning rate at which to end rate decay.
-
-
-
Method Detail
-
setBaseValue
public PolynomialDecayTracker.Builder setBaseValue(float baseValue)
Sets the initial value after no steps.- Parameters:
baseValue- the initial value- Returns:
- this
Builder
-
setEndLearningRate
public PolynomialDecayTracker.Builder setEndLearningRate(float endLearningRate)
Sets the learning rate at which to end rate decay.- Parameters:
endLearningRate- the learning rate at which to end rate decay.- Returns:
- this builder
-
setDecaySteps
public PolynomialDecayTracker.Builder setDecaySteps(int decaySteps)
Sets the number of training steps to decay learning rate in.- Parameters:
decaySteps- the number of training steps to decay learning rate in- Returns:
- this builder
-
optPower
public PolynomialDecayTracker.Builder optPower(float power)
Sets the power of the polynomial to decay by.- Parameters:
power- the power of the polynomial to decay by.- Returns:
- this builder
-
build
public PolynomialDecayTracker build()
Builds a PolynomialDecayTracker.- Returns:
- a PolynomialDecayTracker
-
-