Package ai.djl.training.optimizer

Class AdamW.Builder

java.lang.Object

ai.djl.training.optimizer.Optimizer.OptimizerBuilder<AdamW.Builder>

ai.djl.training.optimizer.AdamW.Builder

Enclosing class:

AdamW

public static final class AdamW.Builder
extends Optimizer.OptimizerBuilder<AdamW.Builder>

The Builder to construct an AdamW object.

Method Summary

Modifier and Type	Method	Description
AdamW	build()	Builds a AdamW block.
AdamW.Builder	optBeta1(float beta1)	Sets the decay rate for the first moment estimates.
AdamW.Builder	optBeta2(float beta2)	Sets the decay rate for the second moment estimates.
AdamW.Builder	optEpsilon(float epsilon)	Sets \(epsilon\) - a small quantity for numerical stability.
AdamW.Builder	optLearningRateTracker(ParameterTracker learningRateTracker)	Sets the ParameterTracker for this optimizer.
protected AdamW.Builder	self()

Methods inherited from class ai.djl.training.optimizer.Optimizer.OptimizerBuilder

optBeginNumUpdate, optClipGrad, optWeightDecays, setRescaleGrad

Methods inherited from class java.lang.Object

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

Method Details

self

protected AdamW.Builder self()

Specified by:

self in class Optimizer.OptimizerBuilder<AdamW.Builder>

optLearningRateTracker

public AdamW.Builder optLearningRateTracker(ParameterTracker learningRateTracker)

Sets the ParameterTracker for this optimizer.

Parameters:

learningRateTracker - the ParameterTracker to be set

Returns:

this Builder

optBeta1

public AdamW.Builder optBeta1(float beta1)

Sets the decay rate for the first moment estimates.

Parameters:

beta1 - the deacay rate for the the first moment estimates

Returns:

this Builder

optBeta2

public AdamW.Builder optBeta2(float beta2)

Sets the decay rate for the second moment estimates.

Parameters:

beta2 - the decay rate for the the second moment estimates

Returns:

this Builder

optEpsilon

public AdamW.Builder optEpsilon(float epsilon)

Sets \(epsilon\) - a small quantity for numerical stability.

Parameters:

epsilon - a small quantity for numerical stability

Returns:

this Builder

build

public AdamW build()

Builds a AdamW block.

Returns:

the AdamW block