RealmOA Library

oa::ml::training::OneCycleScheduler

Smith one-cycle ramp and cosine decay policy.

StRust public API

Declaration

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

Public Methods

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

Associated Functions

pub fn with_defaults(max_learning_rate: f32, total_steps: u64) -> Result<Self>
pub fn new( max_learning_rate: f32, total_steps: u64, percent_start: f32, division_factor: f32, final_division_factor: f32, ) -> Result<Self>

Public Method Documentation

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