Struct Adam 

pub struct Adam {
    pub lr: Float,
    pub beta1: Float,
    pub beta2: Float,
    pub epsilon: Float,
    pub weight_decay: Float,
    /* private fields */
}

Fields

lr: Float

beta1: Float

beta2: Float

epsilon: Float

weight_decay: Float

Implementations

impl Adam

pub fn new(lr: Float) -> Self

pub const fn with_weight_decay(self, weight_decay: Float) -> Self

Trait Implementations

impl Clone for Adam

fn clone(&self) -> Adam

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for Adam

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Optimizer for Adam

fn begin_step(&mut self)

fn update_parameter( &mut self, slot: usize, params: &mut [Float], grads: &[Float], scale: Float, )