Package ai.djl.training.tracker
Class WarmUpTracker
- java.lang.Object
-
- ai.djl.training.tracker.WarmUpTracker
-
- All Implemented Interfaces:
ParameterTracker,Tracker
public final class WarmUpTracker extends java.lang.Object implements Tracker
AWarmUpTrackerapplies a simple warm-up before executing a mainTracker.
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static classWarmUpTracker.BuilderThe Builder to construct aWarmUpTracker.static classWarmUpTracker.ModeAn enum that enumerates the types of warm-up modes for aWarmUpTracker.
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description static WarmUpTracker.Builderbuilder()Creates a new builder.floatgetNewValue(int numUpdate)Fetches the value after the given number of steps/updates.-
Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
-
Methods inherited from interface ai.djl.training.tracker.Tracker
getNewValue
-
-
-
-
Method Detail
-
builder
public static WarmUpTracker.Builder builder()
Creates a new builder.- Returns:
- a new builder
-
getNewValue
public float getNewValue(int numUpdate)
Fetches the value after the given number of steps/updates.- Specified by:
getNewValuein interfaceTracker- Parameters:
numUpdate- the total number of steps/updates- Returns:
- this
Builder
-
-