Optimizer

Trait Optimizer 

Source
pub trait Optimizer<T: Num, D: Device, P: Parameters<T, D>> {
    // Required method
    fn update(&self, parameters: &P);
}

Required Methods§

Source

fn update(&self, parameters: &P)

Implementors§

Source§

impl<T: Num, D: Device, P: Parameters<T, D>> Optimizer<T, D, P> for Adam<T, D>

Source§

impl<T: Num, D: Device, P: Parameters<T, D>> Optimizer<T, D, P> for AdamW<T, D>

Source§

impl<T: Num, D: Device, P: Parameters<T, D>> Optimizer<T, D, P> for SGD<T, D>