pub enum LearningRateSchedule {
Constant {
lr: f32,
},
ExponentialDecay {
initial_lr: f32,
decay_rate: f32,
},
StepDecay {
initial_lr: f32,
step_size: usize,
gamma: f32,
},
CosineAnnealing {
initial_lr: f32,
t_max: usize,
},
WarmRestart {
initial_lr: f32,
t_0: usize,
t_mult: usize,
},
}Expand description
Learning rate scheduling for continual learning
Variants§
Constant
Constant learning rate
ExponentialDecay
Exponential decay
StepDecay
Step decay
CosineAnnealing
Cosine annealing
WarmRestart
Warm restart
Trait Implementations§
Source§impl Clone for LearningRateSchedule
impl Clone for LearningRateSchedule
Source§fn clone(&self) -> LearningRateSchedule
fn clone(&self) -> LearningRateSchedule
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for LearningRateSchedule
impl Debug for LearningRateSchedule
Source§impl<'de> Deserialize<'de> for LearningRateSchedule
impl<'de> Deserialize<'de> for LearningRateSchedule
Source§fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>where
__D: Deserializer<'de>,
Deserialize this value from the given Serde deserializer. Read more
Auto Trait Implementations§
impl Freeze for LearningRateSchedule
impl RefUnwindSafe for LearningRateSchedule
impl Send for LearningRateSchedule
impl Sync for LearningRateSchedule
impl Unpin for LearningRateSchedule
impl UnsafeUnpin for LearningRateSchedule
impl UnwindSafe for LearningRateSchedule
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> ConfigSerializable for Twhere
T: Serialize + for<'de> Deserialize<'de>,
impl<T> ConfigSerializable for Twhere
T: Serialize + for<'de> Deserialize<'de>,
Source§fn save_to_file(&self, path: &Path) -> Result<(), TrustformersError>
fn save_to_file(&self, path: &Path) -> Result<(), TrustformersError>
Save to file
Source§fn load_from_file(path: &Path) -> Result<Self, TrustformersError>where
Self: Sized,
fn load_from_file(path: &Path) -> Result<Self, TrustformersError>where
Self: Sized,
Load from file
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more