ai.djl.training.optimizer.learningrate

Class FactorTracker

java.lang.Object

ai.djl.training.optimizer.learningrate.LearningRateTracker

ai.djl.training.optimizer.learningrate.FactorTracker

public class FactorTracker
extends LearningRateTracker

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

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	FactorTracker.Builder The Builder to construct an FactorTracker 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
FactorTracker(FactorTracker.Builder builder) Creates a new instance of FactorTracker.

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

FactorTracker

public FactorTracker(FactorTracker.Builder builder)

Creates a new instance of FactorTracker.

Parameters:

builder - the builder to create a new instance of FactorTracker

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