Package 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	Description
static class	PolynomialDecayTracker.Builder	Builder for PolynomialDecayTracker.

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method	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

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