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 ATrackerrepresents a hyper-parameter that changes gradually through the training process. -
Class Summary Class Description CosineTracker CosineTrackeris an implementation ofTrackerwhich 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 anCosineTrackerobject.FactorTracker FactorTrackeris an implementation ofTrackerwhich is updated by a multiplicative factor.FactorTracker.Builder The Builder to construct anFactorTrackerobject.LinearTracker FactorTrackeris an implementation ofTrackerwhich is updated by a constant factor.LinearTracker.Builder The Builder to construct anLinearTrackerobject.MultiFactorTracker MultiFactorTrackeris an implementation ofTrackerwhich returns piecewise constant values for fixed numbers of steps.MultiFactorTracker.Builder The Builder to construct anMultiFactorTrackerobject.PolynomialDecayTracker Polynomial decayTracker.PolynomialDecayTracker.Builder Builder for PolynomialDecayTracker.WarmUpTracker AWarmUpTrackerapplies 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.