Package ai.djl.training.tracker

Class FixedPerVarTracker.Builder

java.lang.Object

ai.djl.training.tracker.FixedPerVarTracker.Builder

Enclosing class:

FixedPerVarTracker

public static final class FixedPerVarTracker.Builder
extends Object

The Builder to construct an FixedPerVarTracker object.

Method Summary

Modifier and Type	Method	Description
FixedPerVarTracker	build()	Builds a FixedPerVarTracker block.
FixedPerVarTracker.Builder	put(String parameterId, float value)	Add a kv pair of parameter and its learning rate.
FixedPerVarTracker.Builder	putAll(Map<String,Float> valueMap)	Add kv pairs of parameter and its learning rate.
FixedPerVarTracker.Builder	setDefaultValue(float value)	Set the default learning rate.

Methods inherited from class java.lang.Object

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

Method Details

setDefaultValue

public FixedPerVarTracker.Builder setDefaultValue(float value)

Set the default learning rate.

Parameters:

value - the default learning rate

Returns:

builder

put

public FixedPerVarTracker.Builder put(String parameterId,
 float value)

Add a kv pair of parameter and its learning rate.

Parameters:

parameterId - the parameter id

value - the default learning rate

Returns:

builder

putAll

public FixedPerVarTracker.Builder putAll(Map<String,Float> valueMap)

Add kv pairs of parameter and its learning rate.

Parameters:

valueMap - stores parameterId and learning rate

Returns:

builder

build

public FixedPerVarTracker build()

Builds a FixedPerVarTracker block.

Returns:

the FixedPerVarTracker block