Struct RLScheduler 

pub struct RLScheduler<T: Float + Debug + Send + Sync + 'static> {
    pub initiallr: T,
    pub current_lr: T,
    pub decay_factor: T,
    pub schedule: ScheduleType,
    pub update_count: usize,
    pub schedule_params: HashMap<String, T>,
}

Learning rate scheduling for RL optimizers

Fields

initiallr: T

Initial learning rate

current_lr: T

Current learning rate

decay_factor: T

Decay factor

schedule: ScheduleType

Decay schedule

update_count: usize

Number of updates so far

schedule_params: HashMap<String, T>

Schedule parameters

Implementations

impl<T: Float + Debug + Send + Sync + 'static> RLScheduler<T>

pub fn new(initiallr: T, schedule: ScheduleType) -> Self

Create a new learning rate scheduler

pub fn step(&mut self) -> T

Update learning rate based on schedule

pub fn get_lr(&self) -> T

Get current learning rate

pub fn set_param(&mut self, key: &str, value: T)

Set schedule parameter

Trait Implementations

impl<T: Clone + Float + Debug + Send + Sync + 'static> Clone for RLScheduler<T>

fn clone(&self) -> RLScheduler<T>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<T: Debug + Float + Debug + Send + Sync + 'static> Debug for RLScheduler<T>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.