Struct LearningRateScheduler 

pub struct LearningRateScheduler { /* private fields */ }

Learning rate scheduler with warmup and cosine decay

Implementations

impl LearningRateScheduler

pub fn instance_of<V: NapiRaw>(env: Env, value: V) -> Result<bool>

impl LearningRateScheduler

pub fn into_reference( val: LearningRateScheduler, env: Env, ) -> Result<Reference<LearningRateScheduler>>

pub fn into_instance( self, env: Env, ) -> Result<ClassInstance<LearningRateScheduler>>

impl LearningRateScheduler

pub fn new(initial_lr: f64, warmup_steps: u32, total_steps: u32) -> Self

Create a new learning rate scheduler

Arguments

• initial_lr - Initial/peak learning rate
• warmup_steps - Number of warmup steps
• total_steps - Total training steps

pub fn with_min_lr( initial_lr: f64, warmup_steps: u32, total_steps: u32, min_lr: f64, ) -> Self

Create with minimum learning rate

pub fn get_lr(&self) -> f64

Get learning rate for current step

pub fn step(&mut self) -> f64

Step the scheduler and return current learning rate

pub fn reset(&mut self)

Reset scheduler to initial state

pub fn current_step(&self) -> u32

Get current step

pub fn progress(&self) -> f64

Get progress (0.0 to 1.0)

Trait Implementations

impl FromNapiMutRef for LearningRateScheduler

unsafe fn from_napi_mut_ref( env: napi_env, napi_val: napi_value, ) -> Result<&'static mut Self>

Safety

impl FromNapiRef for LearningRateScheduler

unsafe fn from_napi_ref( env: napi_env, napi_val: napi_value, ) -> Result<&'static Self>

Safety

impl FromNapiValue for &LearningRateScheduler

unsafe fn from_napi_value(env: napi_env, napi_val: napi_value) -> Result<Self>

Safety

fn from_unknown(value: JsUnknown) -> Result<Self, Error>

impl FromNapiValue for &mut LearningRateScheduler

unsafe fn from_napi_value(env: napi_env, napi_val: napi_value) -> Result<Self>

Safety

fn from_unknown(value: JsUnknown) -> Result<Self, Error>

impl ObjectFinalize for LearningRateScheduler

fn finalize(self, env: Env) -> Result<(), Error>

impl ToNapiValue for LearningRateScheduler

unsafe fn to_napi_value( env: napi_env, val: LearningRateScheduler, ) -> Result<napi_value>

Safety

impl TypeName for &LearningRateScheduler

fn type_name() -> &'static str

fn value_type() -> ValueType

impl TypeName for &mut LearningRateScheduler

fn type_name() -> &'static str

fn value_type() -> ValueType

impl TypeName for LearningRateScheduler

fn type_name() -> &'static str

fn value_type() -> ValueType

impl ValidateNapiValue for &LearningRateScheduler

unsafe fn validate(env: napi_env, napi_val: napi_value) -> Result<napi_value>

Safety

impl ValidateNapiValue for &mut LearningRateScheduler

unsafe fn validate(env: napi_env, napi_val: napi_value) -> Result<napi_value>

Safety