Trait hal::optimizer::Optimizer

pub trait Optimizer {
    fn new(params: &HashMap<&str, &str>) -> Self
    where
        Self: Sized;
    fn setup(&mut self, dims: Vec<Dim4>);
    fn update(&mut self, parameter_manager: &mut ParamManager, batch_size: u64);
    fn info(&self);
}

Required Methods

fn new(params: &HashMap<&str, &str>) -> Self where
    Self: Sized, 

fn setup(&mut self, dims: Vec<Dim4>)

fn update(&mut self, parameter_manager: &mut ParamManager, batch_size: u64)

fn info(&self)

Implementors

• impl Optimizer for SGD
• impl Optimizer for Adam