Struct OneCyclePolicyScheduler 

pub struct OneCyclePolicyScheduler { /* private fields */ }

One-cycle learning rate policy.

Phase 1 (pct_start of total_steps): linear ramp from base_lr to max_lr Phase 2 (remaining): cosine decay from max_lr to min_lr

Implementations

impl OneCycleLrScheduler

pub fn new( base_lr: f64, max_lr: f64, min_lr: f64, total_steps: usize, pct_start: f64, ) -> Result<Self, SchedulerError>

Create a new one-cycle learning rate scheduler.

Errors

Returns SchedulerError::InvalidConfig if any parameter is invalid.

Trait Implementations

impl LrSchedulerV2 for OneCycleLrScheduler

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