LRScheduler

quantrs2_ml::pytorch_api

Trait LRScheduler 

Source 
pub trait LRScheduler {
    // Required methods
    fn get_lr(&self) -> f64;
    fn step(&mut self);
    fn set_epoch(&mut self, epoch: usize);
}
Expand description

Learning rate scheduler trait

Required Methods§

Source

fn get_lr(&self) -> f64

Get current learning rate

Source

fn step(&mut self)

Step the scheduler

Source

fn set_epoch(&mut self, epoch: usize)

Set the epoch

Implementors§