Package ai.djl.training.tracker
Class CyclicalTracker
- java.lang.Object
-
- ai.djl.training.tracker.CyclicalTracker
-
- All Implemented Interfaces:
ParameterTracker,Tracker
public class CyclicalTracker extends java.lang.Object implements Tracker
CyclicalTrackeris an implementation ofTrackerwhich is a policy of learning rate adjustment that increases the learning rate off a base value in a cyclical nature, as detailed in the paper Cyclical Learning Rates for Training Neural Networks.
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static classCyclicalTracker.BuilderThe Builder to construct anCyclicalTrackerobject.static classCyclicalTracker.CyclicalModeCyclicalTrackerprovides three predefined cyclical modes and can be selected by this enum.static interfaceCyclicalTracker.ScaleFunctionScaleFunctionis an interface to implement a custom scale function.
-
Constructor Summary
Constructors Constructor Description CyclicalTracker(CyclicalTracker.Builder builder)Creates a new instance ofCyclicalTracker.
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description static CyclicalTracker.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
-
-
-
-
Constructor Detail
-
CyclicalTracker
public CyclicalTracker(CyclicalTracker.Builder builder)
Creates a new instance ofCyclicalTracker.- Parameters:
builder- the builder to create a new instance ofCyclicalTracker
-
-
Method Detail
-
builder
public static CyclicalTracker.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
-
-