RealmOA Library

oa::ml::training::CosineScheduler

Cosine annealing from a maximum rate to a minimum rate.

StRust public API

Declaration

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

Public Methods

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

Associated Functions

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

Public Method Documentation

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