Struct ModuleLrScheduler 

pub struct ModuleLrScheduler { /* private fields */ }

A learning rate scheduler that maps specific parameter groups to dedicated sub-schedulers.

This allows heterogeneous learning rate schedules across different layers of a model (e.g., discriminative layer training or fine-tuning).

Implementations

impl ModuleLrScheduler

pub fn new<S>(scheduler: S) -> ModuleLrScheduler

where S: LrScheduler + 'static,

Create a ModuleLrScheduler.

Arguments

• scheduler - The policy’s default learning rate scheduler.

pub fn step(&mut self) -> ModuleLearningRate

Perform the scheduler step of every scheduler and returns the effective learning rate policy.

pub fn to_record(&self) -> LrSchedulerRecord

Get the current state of the schedulers as a record.

pub fn load_record(self, record: LrSchedulerRecord) -> ModuleLrScheduler

Load the state of the schedulers from a record.

pub fn with_group( self, group: ParamGroup, scheduler: impl Into<DynLrScheduler>, ) -> ModuleLrScheduler

Add a new parameter group to the scheduler’s policy.

Trait Implementations

impl Clone for ModuleLrScheduler

fn clone(&self) -> ModuleLrScheduler

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<S> From<S> for ModuleLrScheduler

where S: LrScheduler + 'static,

fn from(value: S) -> ModuleLrScheduler

Converts to this type from the input type.