Skip to main content

LrScheduler

tensorlogic_train

Trait LrScheduler 

Source 
pub trait LrScheduler {
    // Required methods
    fn step(&mut self, optimizer: &mut dyn Optimizer);
    fn get_lr(&self) -> f64;
    fn state_dict(&self) -> HashMap<String, f64>;
    fn load_state_dict(
        &mut self,
        state: &HashMap<String, f64>,
    ) -> TrainResult<()>;
}
Expand description

Trait for learning rate schedulers.

Required Methods§

Source

fn step(&mut self, optimizer: &mut dyn Optimizer)

Update learning rate based on current step/epoch.

Source

fn get_lr(&self) -> f64

Get current learning rate.

Source

fn state_dict(&self) -> HashMap<String, f64>

Get scheduler state as a dictionary.

Source

fn load_state_dict(&mut self, state: &HashMap<String, f64>) -> TrainResult<()>

Load scheduler state from a dictionary.

Implementors§