Struct ComposedLrSchedulerConfig 

pub struct ComposedLrSchedulerConfig { /* private fields */ }

Compose multiple learning rate schedulers together.

Implementations

impl ComposedLrSchedulerConfig

pub fn new() -> ComposedLrSchedulerConfig

Create a new instance of the config.

Arguments

Default Arguments

schedulers

• Defaults to "Vec::new()"

reduction

• Defaults to "SchedulerReduction::Prod"

impl ComposedLrSchedulerConfig

pub fn with_schedulers( self, schedulers: Vec<LrSchedulerConfig>, ) -> ComposedLrSchedulerConfig

Sets the value for the field schedulers.

• Defaults to "Vec::new()"

pub fn with_reduction( self, reduction: SchedulerReduction, ) -> ComposedLrSchedulerConfig

Sets the value for the field reduction.

• Defaults to "SchedulerReduction::Prod"

impl ComposedLrSchedulerConfig

pub fn init(&self) -> Result<ModuleLrScheduler, String>

Initializes a module learning rate scheduler.

pub fn constant(self, lr: f64) -> ComposedLrSchedulerConfig

Appends a constant learning rate.

pub fn linear( self, config: LinearLrSchedulerConfig, ) -> ComposedLrSchedulerConfig

Appends a linear scheduler.

pub fn cosine( self, config: CosineAnnealingLrSchedulerConfig, ) -> ComposedLrSchedulerConfig

Appends a cosine scheduler.

pub fn exponential( self, config: ExponentialLrSchedulerConfig, ) -> ComposedLrSchedulerConfig

Appends an exponential scheduler.

pub fn noam(self, config: NoamLrSchedulerConfig) -> ComposedLrSchedulerConfig

Appends a noam scheduler.

pub fn step(self, config: StepLrSchedulerConfig) -> ComposedLrSchedulerConfig

Appends a step scheduler.

pub fn composed( self, config: ComposedLrSchedulerConfig, ) -> ComposedLrSchedulerConfig

Appends a composed scheduler.

Trait Implementations

impl Clone for ComposedLrSchedulerConfig

fn clone(&self) -> ComposedLrSchedulerConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Config for ComposedLrSchedulerConfig

fn save<P>(&self, file: P) -> Result<(), Error>

where P: AsRef<Path>,

Saves the configuration to a file.

fn load<P>(file: P) -> Result<Self, ConfigError>

where P: AsRef<Path>,

Loads the configuration from a file.

fn load_binary(data: &[u8]) -> Result<Self, ConfigError>

Loads the configuration from a binary buffer.

impl Debug for ComposedLrSchedulerConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for ComposedLrSchedulerConfig

fn deserialize<D>( deserializer: D, ) -> Result<ComposedLrSchedulerConfig, <D as Deserializer<'de>>::Error>

where D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for ComposedLrSchedulerConfig

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

Formats the value using the given formatter.

impl From<ComposedLrSchedulerConfig> for LrSchedulerConfig

fn from(config: ComposedLrSchedulerConfig) -> LrSchedulerConfig

Converts to this type from the input type.

impl Serialize for ComposedLrSchedulerConfig

fn serialize<S>( &self, serializer: S, ) -> Result<<S as Serializer>::Ok, <S as Serializer>::Error>

where S: Serializer,

Serialize this value into the given Serde serializer.