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 and Description
Nag	build() Builds a Nag block.
protected Nag.Builder	self()
Nag.Builder	setLearningRateTracker(LearningRateTracker learningRateTracker) Sets the LearningRateTracker 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 Detail

setLearningRateTracker

public Nag.Builder setLearningRateTracker(LearningRateTracker learningRateTracker)

Sets the LearningRateTracker for this optimizer.

Parameters:

learningRateTracker - the LearningRateTracker 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