Enum Optimizer 

pub enum Optimizer {
    Adam(Optimizer),
    AdamW(Optimizer),
}

Optimizers.

This is a thin wrapper of tch::nn::Optimizer.

Variants

Adam(Optimizer)

Adam optimizer.

AdamW(Optimizer)

Implementations

impl Optimizer

pub fn backward_step(&mut self, loss: &Tensor)

Applies a backward step pass.