Struct juice::solvers::sgd::momentum::Momentum

pub struct Momentum<SolverB: IBackend + SolverOps<f32>> { /* private fields */ }

Stochastic Gradient Descent with Momentum.

See [module description][1] for more information. [1]: ./index.html

Implementations

impl<SolverB: IBackend + SolverOps<f32>> Momentum<SolverB>

pub fn new(backend: Rc<SolverB>) -> Momentum<SolverB>

Create a new SGD Momentum solver.

Should not be called directly. Use Solver::from_config instead.

Trait Implementations

impl<SolverB: Debug + IBackend + SolverOps<f32>> Debug for Momentum<SolverB>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<SolverB: IBackend + SolverOps<f32>, NetB: IBackend + LayerOps<f32> + 'static> ISolver<SolverB, NetB> for Momentum<SolverB>

fn init(&mut self, net: &Layer<NetB>)

Initialize the SGD Momentum solver, allocating memory for its history.

fn compute_update( &mut self, config: &SolverConfig, net: &mut Layer<NetB>, iter: usize )

Update the weights of the net with part of the gradient.

fn backend(&self) -> &SolverB

Returns the backend used by the solver.