Skip to content

AdamParams

class mint.types.AdamParams ( StrictBase )

Fields:

  • learning_rate ( float ) – Learning rate for the optimizer
  • beta1 ( float ) – Coefficient used for computing running averages of gradient
  • beta2 ( float ) – Coefficient used for computing running averages of gradient square
  • eps ( float ) – Term added to the denominator to improve numerical stability
  • weight_decay ( float ) – Weight decay for the optimizer. Uses decoupled weight decay.
  • grad_clip_norm ( float ) – Maximum global gradient norm. If the global gradient norm is greater than this value, it will be clipped to this value. 0.0 means no clipping.

Referenced by

  • TrainingClient.optim_step