Package ai.djl.training.tracker

Class MultiFactorTracker

java.lang.Object

ai.djl.training.tracker.MultiFactorTracker

All Implemented Interfaces:

ParameterTracker, Tracker

public class MultiFactorTracker
extends java.lang.Object
implements Tracker

MultiFactorTracker is an implementation of Tracker which returns piecewise constant values for fixed numbers of steps. multiplicative factor, at an uneven interval of steps, until it reaches a specified stop value.

See Also:

For tracking learning rates, this section in the D2L chapter on learning rate scheduling

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	MultiFactorTracker.Builder	The Builder to construct an MultiFactorTracker object.

Constructor Summary

Constructors

Constructor	Description
MultiFactorTracker(MultiFactorTracker.Builder builder)	Creates a new instance of MultiFactorTracker.

Method Summary

Modifier and Type	Method	Description
static MultiFactorTracker.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

getNewValue

Constructor Detail

MultiFactorTracker

public MultiFactorTracker(MultiFactorTracker.Builder builder)

Creates a new instance of MultiFactorTracker.

Parameters:

builder - the builder to create a new instance of MultiFactorTracker

Method Detail

builder

public static MultiFactorTracker.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:

getNewValue in interface Tracker

Parameters:

numUpdate - the total number of steps/updates

Returns:

this Builder