ai.djl.training.optimizer.learningrate

Class MultiFactorTracker.Builder

java.lang.Object

ai.djl.training.optimizer.learningrate.LearningRateTracker.LrBaseBuilder<MultiFactorTracker.Builder>

ai.djl.training.optimizer.learningrate.MultiFactorTracker.Builder

Enclosing class:

MultiFactorTracker

public static final class MultiFactorTracker.Builder
extends LearningRateTracker.LrBaseBuilder<MultiFactorTracker.Builder>

The Builder to construct an MultiFactorTracker object.

Method Summary

Modifier and Type	Method and Description
MultiFactorTracker	build() Builds a MultiFactorTracker block.
MultiFactorTracker.Builder	optFactor(float factor) Set the value of the multiplicative factor.
protected MultiFactorTracker.Builder	self()
MultiFactorTracker.Builder	setSteps(int[] steps) Sets an array of integers indicating when the learning rate should be changed, usually in an uneven interval of steps.

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

optBaseLearningRate, optWarmUpBeginLearningRate, optWarmUpMode, optWarmUpSteps

Methods inherited from class java.lang.Object

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

Method Detail

self

protected MultiFactorTracker.Builder self()

Specified by:

self in class LearningRateTracker.LrBaseBuilder<MultiFactorTracker.Builder>

setSteps

public MultiFactorTracker.Builder setSteps(int[] steps)

Sets an array of integers indicating when the learning rate should be changed, usually in an uneven interval of steps.

Parameters:

steps - an array of integers indicating when the learning rate should be change

Returns:

this Builder

optFactor

public MultiFactorTracker.Builder optFactor(float factor)

Set the value of the multiplicative factor.

Parameters:

factor - the value of the multiplicative factor

Returns:

this Builder

build

public MultiFactorTracker build()

Builds a MultiFactorTracker block.

Returns:

the MultiFactorTracker block