public abstract static class LearningRateTracker.LrBaseBuilder<T extends LearningRateTracker.LrBaseBuilder>
extends java.lang.Object
LearningRateTracker
.Constructor and Description |
---|
LrBaseBuilder() |
Modifier and Type | Method and Description |
---|---|
T |
optBaseLearningRate(float baseLearningRate)
Sets the base learning rate.
|
T |
optWarmUpBeginLearningRate(float warmUpBeginLearningRate)
Sets the value of the learning rate at the beginning of warm-up mode.
|
T |
optWarmUpMode(WarmUpMode warmUpMode)
Sets the
WarmUpMode for the LearningRateTracker . |
T |
optWarmUpSteps(int warmUpSteps)
Sets the number of steps until the point the learning rate is updated in warm-up mode.
|
protected abstract T |
self() |
public T optBaseLearningRate(float baseLearningRate)
baseLearningRate
- the base learning rateBuilder
public T optWarmUpSteps(int warmUpSteps)
warmUpSteps
- the number of steps until the point the learning rate is updated in
warm-up modeBuilder
public T optWarmUpBeginLearningRate(float warmUpBeginLearningRate)
warmUpBeginLearningRate
- the value of the learning rate at the beginning of warm-up
modeBuilder
public T optWarmUpMode(WarmUpMode warmUpMode)
WarmUpMode
for the LearningRateTracker
.warmUpMode
- the WarmUpMode
to be setBuilder
protected abstract T self()