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.Object
Builder for PolynomialDecayTracker.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description PolynomialDecayTracker
build()
Builds a PolynomialDecayTracker.PolynomialDecayTracker.Builder
optPower(float power)
Sets the power of the polynomial to decay by.PolynomialDecayTracker.Builder
setBaseValue(float baseValue)
Sets the initial value after no steps.PolynomialDecayTracker.Builder
setDecaySteps(int decaySteps)
Sets the number of training steps to decay learning rate in.PolynomialDecayTracker.Builder
setEndLearningRate(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
-
-