Struct ReduceLROnPlateau

pub struct ReduceLROnPlateau { /* private fields */ }

Reduce learning rate on plateau (when validation loss stops improving)

Implementations

impl ReduceLROnPlateau

pub fn new(factor: f64, patience: usize) -> Self

pub fn with_params( factor: f64, patience: usize, threshold: f64, cooldown: usize, min_lr: f64, ) -> Self

pub fn step(&mut self, val_loss: f64, base_lr: f64) -> f64

Update the scheduler with the current validation loss

Trait Implementations

impl Clone for ReduceLROnPlateau

fn clone(&self) -> ReduceLROnPlateau

Returns a duplicate of the value.

const fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ReduceLROnPlateau

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl LearningRateScheduler for ReduceLROnPlateau

fn get_lr(&mut self, _epoch: usize, base_lr: f64) -> f64

Get the learning rate for the current epoch

fn reset(&mut self)

Reset the scheduler state (useful for multiple training runs)

fn name(&self) -> &'static str

Get the name of the scheduler for logging