Trait auto_diff::optim::Optimizer

pub trait Optimizer {
    fn step(&mut self, m: &Module);
    fn step2(&mut self, m: &Func);
}

Required methods

fn step(&mut self, m: &Module)

fn step2(&mut self, m: &Func)

Implementors

impl Optimizer for SGD