public static final class Adam.Builder extends Optimizer.OptimizerBuilder<Adam.Builder>
Adam
object.Modifier and Type | Method and Description |
---|---|
Adam |
build()
Builds a
Adam block. |
Adam.Builder |
optBeta1(float beta1)
Sets the decay rate for the first moment estimates.
|
Adam.Builder |
optBeta2(float beta2)
Sets the decay rate for the second moment estimates.
|
Adam.Builder |
optEpsilon(float epsilon)
Sets \(epsilon\) - a small quantity for numerical stability.
|
Adam.Builder |
optLearningRateTracker(Tracker learningRateTracker)
Sets the
Tracker for this optimizer. |
protected Adam.Builder |
self() |
optBeginNumUpdate, optClipGrad, optWeightDecays, setRescaleGrad
protected Adam.Builder self()
self
in class Optimizer.OptimizerBuilder<Adam.Builder>
public Adam.Builder optLearningRateTracker(Tracker learningRateTracker)
Tracker
for this optimizer.learningRateTracker
- the Tracker
to be setBuilder
public Adam.Builder optBeta1(float beta1)
beta1
- the deacay rate for the the first moment estimatesBuilder
public Adam.Builder optBeta2(float beta2)
beta2
- the decay rate for the the second moment estimatesBuilder
public Adam.Builder optEpsilon(float epsilon)
epsilon
- a small quantity for numerical stabilityBuilder