public abstract class LearningRateTracker
extends java.lang.Object
LearningRateTracker
tracks the evolution of the learning rate through the training
process.Modifier and Type | Class and Description |
---|---|
static class |
LearningRateTracker.LrBaseBuilder<T extends LearningRateTracker.LrBaseBuilder>
The Builder to construct a
LearningRateTracker . |
Modifier and Type | Method and Description |
---|---|
static FactorTracker.Builder |
factorTracker()
Returns a new instance of
FactorTracker.Builder that can build an FactorTracker . |
static ai.djl.training.optimizer.learningrate.FixedLearningRate |
fixedLearningRate(float learningRate)
Returns a new instance of
MultiFactorTracker . |
abstract float |
getNewLearningRate(int numUpdate)
Fetches the value of the learning rate after the given number of steps/updates.
|
static MultiFactorTracker.Builder |
multiFactorTracker()
Returns a new instance of
MultiFactorTracker.Builder that can build an MultiFactorTracker . |
public abstract float getNewLearningRate(int numUpdate)
numUpdate
- the number of steps/updatesBuilder
public static FactorTracker.Builder factorTracker()
FactorTracker.Builder
that can build an FactorTracker
.FactorTracker
FactorTracker.Builder
public static MultiFactorTracker.Builder multiFactorTracker()
MultiFactorTracker.Builder
that can build an MultiFactorTracker
.MultiFactorTracker
MultiFactorTracker.Builder
public static ai.djl.training.optimizer.learningrate.FixedLearningRate fixedLearningRate(float learningRate)
MultiFactorTracker
.learningRate
- the fixed learning rateMultiFactorTracker
MultiFactorTracker.Builder