pub trait LearningRateSchedule: Send + Sync { // Required method fn get_lr(&self, step: usize) -> f64; }
Learning rate schedule trait
Get learning rate for given step