Package ai.djl.training.tracker
Contains classes for having a gradually changing hyper-parameter.
It contains a main interface Tracker
and various options that
extend it.
-
Interface Summary Interface Description Tracker ATracker
represents a hyper-parameter that changes gradually through the training process. -
Class Summary Class Description CosineTracker CosineTracker
is an implementation ofTracker
which is updated by taking sections of a cosine curve to smoothly reduce learning rate until a specified step and base learning rate.CosineTracker.Builder The Builder to construct anCosineTracker
object.FactorTracker FactorTracker
is an implementation ofTracker
which is updated by a multiplicative factor.FactorTracker.Builder The Builder to construct anFactorTracker
object.LinearTracker FactorTracker
is an implementation ofTracker
which is updated by a constant factor.LinearTracker.Builder The Builder to construct anLinearTracker
object.MultiFactorTracker MultiFactorTracker
is an implementation ofTracker
which returns piecewise constant values for fixed numbers of steps.MultiFactorTracker.Builder The Builder to construct anMultiFactorTracker
object.PolynomialDecayTracker Polynomial decayTracker
.PolynomialDecayTracker.Builder Builder for PolynomialDecayTracker.WarmUpTracker AWarmUpTracker
applies a simple warm-up before executing a mainTracker
.WarmUpTracker.Builder The Builder to construct aWarmUpTracker
. -
Enum Summary Enum Description WarmUpTracker.Mode An enum that enumerates the types of warm-up modes for aWarmUpTracker
.