Package ai.djl.training.optimizer

Class Nag.Builder

java.lang.Object

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

ai.djl.training.optimizer.Nag.Builder

Enclosing class:

Nag

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

The Builder to construct an Nag object.

Method Summary

Modifier and Type	Method	Description
Nag	build()	Builds a Nag block.
protected Nag.Builder	self()
Nag.Builder	setLearningRateTracker(ParameterTracker learningRateTracker)	Sets the ParameterTracker for this optimizer.
Nag.Builder	setMomentum(float momentum)	Sets the momentum for Nag.

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

setLearningRateTracker

public Nag.Builder setLearningRateTracker(ParameterTracker learningRateTracker)

Sets the ParameterTracker for this optimizer.

Parameters:

learningRateTracker - the ParameterTracker to be set

Returns:

this Builder

setMomentum

public Nag.Builder setMomentum(float momentum)

Sets the momentum for Nag.

Parameters:

momentum - the value of momentum

Returns:

this Builder

self

protected Nag.Builder self()

Specified by:

self in class Optimizer.OptimizerBuilder<Nag.Builder>

build

public Nag build()

Builds a Nag block.

Returns:

the Nag block