ai.djl.training.optimizer.learningrate

Class MultiFactorTracker

java.lang.Object

ai.djl.training.optimizer.learningrate.LearningRateTracker

ai.djl.training.optimizer.learningrate.MultiFactorTracker

public class MultiFactorTracker
extends LearningRateTracker

MultiFactorTracker is an implementation of LearningRateTracker which is updated by a multiplicative factor, at an uneven interval of steps, until it reaches a specified stop value.

Nested Class Summary

Nested Classes

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

Nested classes/interfaces inherited from class ai.djl.training.optimizer.learningrate.LearningRateTracker

LearningRateTracker.LrBaseBuilder<T extends LearningRateTracker.LrBaseBuilder>

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
float	getNewLearningRate(int numUpdate) Fetches the value of the learning rate after the given number of steps/updates.

Methods inherited from class ai.djl.training.optimizer.learningrate.LearningRateTracker

factorTracker, fixedLearningRate, multiFactorTracker

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

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

getNewLearningRate

public float getNewLearningRate(int numUpdate)

Fetches the value of the learning rate after the given number of steps/updates.

Specified by:

getNewLearningRate in class LearningRateTracker

Parameters:

numUpdate - the number of steps/updates

Returns:

this Builder