pub struct ProgressiveConfig {
pub growth_dimension: GrowthDimension,
pub growth_strategy: GrowthStrategy,
pub initial_size: usize,
pub final_size: usize,
pub growth_epochs: Vec<usize>,
pub warmup_steps: usize,
pub zero_init_new_params: bool,
pub lr_scaling_factor: f64,
pub gradual_initialization: bool,
pub transition_smoothing: f64,
pub freeze_old_params_during_warmup: bool,
}Expand description
Configuration for progressive training
Fields§
§growth_dimension: GrowthDimensionWhich dimension to grow progressively
growth_strategy: GrowthStrategyGrowth strategy to use
initial_size: usizeInitial model size (layers, hidden dim, heads, etc.)
final_size: usizeFinal model size
growth_epochs: Vec<usize>Epochs at which to trigger growth
warmup_steps: usizeSteps to warm up after each growth
zero_init_new_params: boolWhether to initialize new parameters with zeros
lr_scaling_factor: f64Learning rate scaling factor after growth
gradual_initialization: boolWhether to use gradual weight initialization
transition_smoothing: f64Smoothing factor for parameter transitions
freeze_old_params_during_warmup: boolWhether to freeze old parameters during warmup
Trait Implementations§
Source§impl Clone for ProgressiveConfig
impl Clone for ProgressiveConfig
Source§fn clone(&self) -> ProgressiveConfig
fn clone(&self) -> ProgressiveConfig
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 ProgressiveConfig
impl Debug for ProgressiveConfig
Source§impl Default for ProgressiveConfig
impl Default for ProgressiveConfig
Source§impl<'de> Deserialize<'de> for ProgressiveConfig
impl<'de> Deserialize<'de> for ProgressiveConfig
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 ProgressiveConfig
impl RefUnwindSafe for ProgressiveConfig
impl Send for ProgressiveConfig
impl Sync for ProgressiveConfig
impl Unpin for ProgressiveConfig
impl UnsafeUnpin for ProgressiveConfig
impl UnwindSafe for ProgressiveConfig
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