ai.djl.training.tracker

Class PolynomialDecayTracker.Builder

java.lang.Object

ai.djl.training.tracker.PolynomialDecayTracker.Builder

Enclosing class:

PolynomialDecayTracker

public static final class PolynomialDecayTracker.Builder
extends java.lang.Object

Builder for PolynomialDecayTracker.

Method Summary

Modifier and Type	Method and Description
PolynomialDecayTracker	build() Builds a PolynomialDecayTracker.
PolynomialDecayTracker.Builder	optPower(float power) Sets the power of the polynomial to decay by.
PolynomialDecayTracker.Builder	setBaseValue(float baseValue) Sets the initial value after no steps.
PolynomialDecayTracker.Builder	setDecaySteps(int decaySteps) Sets the number of training steps to decay learning rate in.
PolynomialDecayTracker.Builder	setEndLearningRate(float endLearningRate) Sets the learning rate at which to end rate decay.

Methods inherited from class java.lang.Object

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

Method Detail

setBaseValue

public PolynomialDecayTracker.Builder setBaseValue(float baseValue)

Sets the initial value after no steps.

Parameters:

baseValue - the initial value

Returns:

this Builder

setEndLearningRate

public PolynomialDecayTracker.Builder setEndLearningRate(float endLearningRate)

Sets the learning rate at which to end rate decay.

Parameters:

endLearningRate - the learning rate at which to end rate decay.

Returns:

this builder

setDecaySteps

public PolynomialDecayTracker.Builder setDecaySteps(int decaySteps)

Sets the number of training steps to decay learning rate in.

Parameters:

decaySteps - the number of training steps to decay learning rate in

Returns:

this builder

optPower

public PolynomialDecayTracker.Builder optPower(float power)

Sets the power of the polynomial to decay by.

Parameters:

power - the power of the polynomial to decay by.

Returns:

this builder

build

public PolynomialDecayTracker build()

Builds a PolynomialDecayTracker.

Returns:

a PolynomialDecayTracker