Struct StepLR 

pub struct StepLR { /* private fields */ }

Multiply the learning rate by gamma every step_size steps.

lr = initial_lr * gamma^(current_step / step_size)

Example

let mut sched = StepLR::new(0.1, 30, 0.1); // decay by 10x every 30 steps

Implementations

impl StepLR

pub fn new(initial_lr: f64, step_size: u64, gamma: f64) -> StepLR

Trait Implementations

impl LrScheduler for StepLR

fn step(&mut self) -> f64

Advance by one step and return the new learning rate.

fn current_lr(&self) -> f64

Get the current learning rate without advancing.

fn current_step(&self) -> u64

Get the current step count.

fn reset(&mut self)

Reset the scheduler to step 0.

fn set_step(&mut self, step: u64)

Set the internal step counter to a specific value (for checkpoint restore).