rstorch/optim/
mod.rs

1use crate::prelude::Module;
2
3mod sgd;
4pub use sgd::SGD;
5
6pub trait Optimizer {
7    fn step<M: Module>(&mut self, module: &mut M);
8}