Package ai.djl.training.tracker
Class WarmUpTracker.Builder
- java.lang.Object
-
- ai.djl.training.tracker.WarmUpTracker.Builder
-
- Enclosing class:
- WarmUpTracker
public static final class WarmUpTracker.Builder extends java.lang.ObjectThe Builder to construct aWarmUpTracker.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description WarmUpTrackerbuild()Builds aWarmUpTrackerblock.WarmUpTracker.BuilderoptWarmUpBeginValue(float warmUpBeginValue)Sets the value at the beginning of warm-up mode.WarmUpTracker.BuilderoptWarmUpMode(WarmUpTracker.Mode warmUpMode)Sets theWarmUpTracker.Modefor theWarmUpTracker.WarmUpTracker.BuilderoptWarmUpSteps(int warmUpSteps)Sets the number of steps until the point the value is updated in warm-up mode.WarmUpTracker.BuildersetMainTracker(Tracker mainTracker)Sets the base value.
-
-
-
Method Detail
-
setMainTracker
public WarmUpTracker.Builder setMainTracker(Tracker mainTracker)
Sets the base value.- Parameters:
mainTracker- the tracker to use after warm up ends- Returns:
- this
Builder
-
optWarmUpSteps
public WarmUpTracker.Builder optWarmUpSteps(int warmUpSteps)
Sets the number of steps until the point the value is updated in warm-up mode.- Parameters:
warmUpSteps- the number of steps the value is updated in warm-up mode- Returns:
- this
Builder
-
optWarmUpBeginValue
public WarmUpTracker.Builder optWarmUpBeginValue(float warmUpBeginValue)
Sets the value at the beginning of warm-up mode.- Parameters:
warmUpBeginValue- the value at the beginning of warm-up mode- Returns:
- this
Builder
-
optWarmUpMode
public WarmUpTracker.Builder optWarmUpMode(WarmUpTracker.Mode warmUpMode)
Sets theWarmUpTracker.Modefor theWarmUpTracker.- Parameters:
warmUpMode- theWarmUpTracker.Modeto be set- Returns:
- this
Builder
-
build
public WarmUpTracker build()
Builds aWarmUpTrackerblock.- Returns:
- the
WarmUpTrackerblock
-
-