scirs2_neural::callbacks

Trait LearningRateScheduler

Source 
pub trait LearningRateScheduler<F: Float + Debug + ScalarOperand> {
    // Required method
    fn get_learning_rate(&mut self, progress: f64) -> Result<F>;

    // Provided method
    fn reset(&mut self) { ... }
}
Expand description

Trait for learning rate schedulers

Required Methods§

Source

fn get_learning_rate(&mut self, progress: f64) -> Result<F>

Get the learning rate for the current progress level (0.0 to 1.0)

Provided Methods§

Source

fn reset(&mut self)

Reset the scheduler state

Implementors§