Struct burn::lr_scheduler::noam::NoamLrScheduler

pub struct NoamLrScheduler { /* private fields */ }

Noam learning rate scheduler as described in Attention Is All You Need.

Trait Implementations

impl Clone for NoamLrScheduler

fn clone(&self) -> NoamLrScheduler

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for NoamLrScheduler

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

Formats the value using the given formatter.

impl LrScheduler for NoamLrScheduler

type Record<B: Backend> = usize

Scheduler associative type to be used when saving and loading the state.

fn step(&mut self) -> f64

Perform the scheduler step, potentially updating its state, and returning the effective learning rate.

fn to_record<B>(&self) -> <NoamLrScheduler as LrScheduler>::Record<B>

where B: Backend,

Get the current state of the scheduler as a record.

fn load_record<B>( self, record: <NoamLrScheduler as LrScheduler>::Record<B>, ) -> NoamLrScheduler

where B: Backend,

Load the state of the scheduler as a record.