ai.djl.training.tracker

Class CosineTracker

java.lang.Object

ai.djl.training.tracker.CosineTracker

All Implemented Interfaces:

Tracker

public class CosineTracker
extends java.lang.Object
implements Tracker

CosineTracker is an implementation of Tracker which is updated by taking sections of a cosine curve to smoothly reduce learning rate until a specified step and base learning rate.

Nested Class Summary

Nested Classes

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

Constructor Summary

Constructors

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

Method Summary

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

CosineTracker

public CosineTracker(CosineTracker.Builder builder)

Creates a new instance of CosineTracker.

Parameters:

builder - the builder to create a new instance of CosineTracker

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