Struct PolynomialLR 

pub struct PolynomialLR { /* private fields */ }

Polynomial learning rate decay

Implementations

impl PolynomialLR

pub fn new(total_iters: usize, power: f64, end_lr: f64) -> Self

Trait Implementations

impl Clone for PolynomialLR

fn clone(&self) -> PolynomialLR

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PolynomialLR

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

Formats the value using the given formatter.

impl LearningRateScheduler for PolynomialLR

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