Package ai.djl.training.tracker
Class PolynomialDecayTracker
- java.lang.Object
-
- ai.djl.training.tracker.PolynomialDecayTracker
-
- All Implemented Interfaces:
ParameterTracker,Tracker
public class PolynomialDecayTracker extends java.lang.Object implements Tracker
Polynomial decayTracker.- See Also:
Tracker
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static classPolynomialDecayTracker.BuilderBuilder for PolynomialDecayTracker.
-
Constructor Summary
Constructors Constructor Description PolynomialDecayTracker(PolynomialDecayTracker.Builder builder)Builds a PolynomialDecayTracker.
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Modifier and Type Method Description static PolynomialDecayTracker.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
-
PolynomialDecayTracker
public PolynomialDecayTracker(PolynomialDecayTracker.Builder builder)
Builds a PolynomialDecayTracker.- Parameters:
builder- parameters
-
-
Method Detail
-
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
-
builder
public static PolynomialDecayTracker.Builder builder()
Creates a new builder.- Returns:
- a new builder
-
-