ai.djl.training.tracker

Class PolynomialDecayTracker

java.lang.Object

ai.djl.training.tracker.PolynomialDecayTracker

All Implemented Interfaces:

Tracker

public class PolynomialDecayTracker
extends java.lang.Object
implements Tracker

Polynomial decay Tracker.

See Also:

Tracker

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	PolynomialDecayTracker.Builder Builder for PolynomialDecayTracker.

Constructor Summary

Constructors

Constructor and Description
PolynomialDecayTracker(PolynomialDecayTracker.Builder builder) Builds a PolynomialDecayTracker.

Method Summary

Modifier and Type	Method and Description
static PolynomialDecayTracker.Builder	builder() Creates a new builder.
float	getNewValue(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

cosine, factor, fixed, multiFactor, warmUp

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:

getNewValue in interface Tracker

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