AdamW.Builder (Deep Java Library 0.24.0 API specification)

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

All Methods Instance Methods Concrete Methods
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 Detail
  - 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