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
pub trait Optimizer {
// Required methods
fn step(
&mut self,
params: &mut HashMap<String, f64>,
gradients: &HashMap<String, f64>,
);
fn reset(&mut self);
}
Base optimizer trait