Trait LRScheduler

pub trait LRScheduler {
    // Required methods
    fn get_lr(&self) -> f32;
    fn step(&mut self);
}

Expand description

Learning rate scheduler trait

Required Methods§

Get the current learning rate

Step the scheduler (typically called after each epoch or batch)

This trait is dyn compatible.

In older versions of Rust, dyn compatibility was called "object safety".