Package ai.djl.training.tracker

Class MultiFactorTracker.Builder

java.lang.Object

ai.djl.training.tracker.MultiFactorTracker.Builder

Enclosing class:

MultiFactorTracker

public static final class MultiFactorTracker.Builder
extends Object

The Builder to construct an MultiFactorTracker object.

Method Summary

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

Methods inherited from class java.lang.Object

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

Method Details

setBaseValue

public MultiFactorTracker.Builder setBaseValue(float baseValue)

Sets the initial value after no steps.

Parameters:

baseValue - the initial value

Returns:

this Builder

setSteps

public MultiFactorTracker.Builder setSteps(int[] steps)

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

Parameters:

steps - an array of integers indicating when the value 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