Struct WarmupScheduler 

pub struct WarmupScheduler { /* private fields */ }

Linear warmup followed by another scheduler.

During warmup: lr = warmup_start_lr + (warmup_end_lr - warmup_start_lr) * (step / warmup_steps) After warmup: delegates to the inner scheduler.

Implementations

impl WarmupScheduler

pub fn new( warmup_steps: usize, warmup_start_lr: f64, warmup_end_lr: f64, inner: Box<dyn LrSchedulerV2>, ) -> Result<Self, SchedulerError>

Create a new warmup scheduler wrapping an inner scheduler.

Errors

Returns SchedulerError::InvalidConfig if warmup_steps is zero.

Trait Implementations

impl LrSchedulerV2 for WarmupScheduler

fn step(&mut self) -> f64

Advance one step and return the new learning rate.

fn current_lr(&self) -> f64

Return the current learning rate without advancing.

fn reset(&mut self)

Reset the scheduler to its initial state.

fn steps_taken(&self) -> usize

Total number of steps taken.

fn completed_cycle(&self) -> bool

Whether the scheduler has completed a cycle (if applicable).