RealmOA Library

oa::ml::training::LinearWarmupCosineScheduler

Linear warmup followed by cosine annealing.

StRust public API

Declaration

pub struct LinearWarmupCosineScheduler { /* private fields */ }
Public Rust declaration.

Public Methods

fn learning_rate(&self, step: u64) -> f32

Associated Functions

pub fn new( warmup_steps: u64, total_steps: u64, max_learning_rate: f32, min_learning_rate: f32, ) -> Result<Self>

Public Method Documentation

fn learning_rate( &self, step: u64 ) -> f32
No public source comment is attached to this declaration.
pub fn new( warmup_steps: u64, total_steps: u64, max_learning_rate: f32, min_learning_rate: f32 ) -> Result<Self>
No public source comment is attached to this declaration.