1 2 3 4 5 6 7 8 9 10
use rai_core::Tensor; use std::collections::HashMap; pub trait Optimizer { /// return new value for the parameters fn step(&mut self, grads: &HashMap<usize, Tensor>) -> HashMap<usize, Tensor>; } mod sdg; pub use sdg::SDG;