Struct StepLR

pub struct StepLR { /* private fields */ }

Step decay scheduler: multiply LR by gamma every step_size epochs

Implementations

impl StepLR

pub fn new(step_size: usize, gamma: f64) -> Self

Trait Implementations

impl Clone for StepLR

fn clone(&self) -> StepLR

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for StepLR

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl LearningRateScheduler for StepLR

fn get_lr(&mut self, epoch: usize, base_lr: f64) -> f64

Get the learning rate for the current epoch

fn reset(&mut self)

Reset the scheduler state (useful for multiple training runs)

fn name(&self) -> &'static str

Get the name of the scheduler for logging