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.Object
The Builder to construct aWarmUpTracker
.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description WarmUpTracker
build()
Builds aWarmUpTracker
block.WarmUpTracker.Builder
optWarmUpBeginValue(float warmUpBeginValue)
Sets the value at the beginning of warm-up mode.WarmUpTracker.Builder
optWarmUpMode(WarmUpTracker.Mode warmUpMode)
Sets theWarmUpTracker.Mode
for theWarmUpTracker
.WarmUpTracker.Builder
optWarmUpSteps(int warmUpSteps)
Sets the number of steps until the point the value is updated in warm-up mode.WarmUpTracker.Builder
setMainTracker(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.Mode
for theWarmUpTracker
.- Parameters:
warmUpMode
- theWarmUpTracker.Mode
to be set- Returns:
- this
Builder
-
build
public WarmUpTracker build()
Builds aWarmUpTracker
block.- Returns:
- the
WarmUpTracker
block
-
-