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)

Loading content...

Implementors

impl Optimizer for SGD

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

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

Loading content...