Skip to main content

Optimizer

Trait Optimizer 

Source
pub trait Optimizer: Debug {
    // Required method
    fn update_parameter(
        &mut self,
        slot: usize,
        params: &mut [Float],
        grads: &[Float],
        scale: Float,
    );

    // Provided method
    fn begin_step(&mut self) { ... }
}

Required Methods§

Source

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

Provided Methods§

Source

fn begin_step(&mut self)

Implementors§