Package ai.djl.training.optimizer

Class Adagrad.Builder

java.lang.Object

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

ai.djl.training.optimizer.Adagrad.Builder

Enclosing class:

Adagrad

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

The Builder to construct an Adagrad object.

Method Summary

Modifier and Type	Method	Description
Adagrad	build()	Builds a Adagrad block.
Adagrad.Builder	optEpsilon(float epsilon)	Sets \(epsilon\) - a small quantity for numerical stability.
Adagrad.Builder	optLearningRateTracker(ParameterTracker learningRateTracker)	Sets the ParameterTracker for this optimizer.
protected Adagrad.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 Adagrad.Builder self()

Specified by:

self in class Optimizer.OptimizerBuilder<Adagrad.Builder>

optLearningRateTracker

public Adagrad.Builder optLearningRateTracker(ParameterTracker learningRateTracker)

Sets the ParameterTracker for this optimizer.

Parameters:

learningRateTracker - the ParameterTracker to be set

Returns:

this Builder

optEpsilon

public Adagrad.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 Adagrad build()

Builds a Adagrad block.

Returns:

the Adagrad block