Optimizer

Trait Optimizer 

Source
pub trait Optimizer {
    // Required methods
    fn step(
        &mut self,
        params: &mut HashMap<String, f64>,
        gradients: &HashMap<String, f64>,
    );
    fn reset(&mut self);
}
Expand description

Base optimizer trait

Required Methods§

Source

fn step( &mut self, params: &mut HashMap<String, f64>, gradients: &HashMap<String, f64>, )

Update parameters given gradients

Source

fn reset(&mut self)

Reset optimizer state

Implementors§