Struct MultiStepLR 

pub struct MultiStepLR { /* private fields */ }

Multi-step decay: multiply lr by gamma at each milestone.

lr = base_lr * gamma^(number_of_milestones_passed)

let sched = MultiStepLR::new(0.1, &[30, 60, 90], 0.1);
assert!((sched.lr(0) - 0.1).abs() < 1e-10);
assert!((sched.lr(30) - 0.01).abs() < 1e-10);
assert!((sched.lr(60) - 0.001).abs() < 1e-10);

Implementations

impl MultiStepLR

pub fn new(base_lr: f64, milestones: &[usize], gamma: f64) -> Self

Create a multi-step scheduler that decays lr by gamma at each milestone epoch. Milestones should be provided in ascending order.

pub fn lr(&self, step: usize) -> f64

Learning rate at the given step.

Trait Implementations

impl Scheduler for MultiStepLR

fn lr(&self, step: usize) -> f64

Learning rate at the given step (0-based).