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 java.lang.ObjectThe Builder to construct anFixedPerVarTrackerobject.
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description FixedPerVarTrackerbuild()Builds aFixedPerVarTrackerblock.FixedPerVarTracker.Builderput(java.lang.String parameterId, float value)Add a kv pair of parameter and its learning rate.FixedPerVarTracker.BuilderputAll(java.util.Map<java.lang.String,java.lang.Float> valueMap)Add kv pairs of parameter and its learning rate.FixedPerVarTracker.BuildersetDefaultValue(float value)Set the default learning rate.
-
-
-
Method Detail
-
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(java.lang.String parameterId, float value)
Add a kv pair of parameter and its learning rate.- Parameters:
parameterId- the parameter idvalue- the default learning rate- Returns:
- builder
-
putAll
public FixedPerVarTracker.Builder putAll(java.util.Map<java.lang.String,java.lang.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 aFixedPerVarTrackerblock.- Returns:
- the
FixedPerVarTrackerblock
-
-