Trait OptimizerStep

pub trait OptimizerStep<F: Float + Debug + ScalarOperand> {
    // Required method
    fn step<L: ParamLayer<F> + ?Sized>(&mut self, model: &mut L) -> Result<()>;
}

Extension trait for optimizers that can work with model layers

Required Methods

fn step<L: ParamLayer<F> + ?Sized>(&mut self, model: &mut L) -> Result<()>

Update model parameters using the optimizer

Dyn Compatibility

This trait is not dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety", so this trait is not object safe.

Implementors

impl<F: Float + Debug + ScalarOperand> OptimizerStep<F> for dyn Optimizer<F> + Send + Sync

Implementation for trait objects

impl<F: Float + Debug + ScalarOperand, O: Optimizer<F>> OptimizerStep<F> for O

Blanket implementation for all optimizers