Skip to main content

StepOptimizer

yscv_optim

Trait StepOptimizer 

Source 
pub trait StepOptimizer {
    // Required method
    fn step(
        &mut self,
        parameter_id: u64,
        weights: &mut Tensor,
        grad: &Tensor,
    ) -> Result<(), OptimError>;
}
Expand description

Trait for optimizers that support a per-parameter step update.

Required Methods§

Source

fn step( &mut self, parameter_id: u64, weights: &mut Tensor, grad: &Tensor, ) -> Result<(), OptimError>

Implementors§