oa::ml::training::WarmupScheduler

Linear warmup followed by an optional delegated schedule.

StRust public API

Declaration

pub struct WarmupScheduler { /* private fields */ }

Public Rust declaration.

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

pub fn new(target_learning_rate: f32, warmup_steps: u64) -> Result<Self>

pub fn with_after( target_learning_rate: f32, warmup_steps: u64, after: Box<dyn LrScheduler>, ) -> Result<Self>

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

No public source comment is attached to this declaration.

pub fn new(
	target_learning_rate: f32,
	warmup_steps: u64
) -> Result<Self>

No public source comment is attached to this declaration.

pub fn with_after(
	target_learning_rate: f32,
	warmup_steps: u64,
	after: Box<dyn LrScheduler>
) -> Result<Self>

No public source comment is attached to this declaration.