Trait auto_diff::optim::Optimizer

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

Required methods

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

Loading content...

Implementors

impl Optimizer for SGD

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

Loading content...