oa::ml::training::CosineWarmRestartsScheduler

Cosine annealing with periodic warm restarts.

StRust public API

Declaration

pub struct CosineWarmRestartsScheduler { /* private fields */ }

Public Rust declaration.

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

pub fn fixed(max_learning_rate: f32, initial_period: u64) -> Result<Self>

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

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

No public source comment is attached to this declaration.

pub fn fixed(
	max_learning_rate: f32,
	initial_period: u64
) -> Result<Self>

No public source comment is attached to this declaration.

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

No public source comment is attached to this declaration.