Skip to main content

TensorFlowLearningRateSchedule

trustformers_optim::tensorflow_compat

Trait TensorFlowLearningRateSchedule 

Source 
pub trait TensorFlowLearningRateSchedule: Send + Sync {
    // Required methods
    fn get_lr(&self, step: i64) -> f64;
    fn get_config(&self) -> Value;
}
Expand description

TensorFlow-compatible learning rate schedule

Required Methods§

Source

fn get_lr(&self, step: i64) -> f64

Get learning rate at current step

Source

fn get_config(&self) -> Value

Get configuration

Implementors§