Struct CosineAnnealingLR 

pub struct CosineAnnealingLR { /* private fields */ }

Cosine annealing from initial_lr to min_lr over total_steps.

lr = min_lr + 0.5 * (initial_lr - min_lr) * (1 + cos(π * step / total_steps))

After total_steps, the LR stays at min_lr.

Implementations

impl CosineAnnealingLR

pub fn new(initial_lr: f64, total_steps: u64, min_lr: f64) -> CosineAnnealingLR

Trait Implementations

impl LrScheduler for CosineAnnealingLR

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).