pub struct AMacPConfig {
pub learning_rate: f32,
pub beta1: f32,
pub beta2: f32,
pub gamma: f32,
pub alpha: f32,
pub eta: f32,
pub epsilon: f32,
pub weight_decay: f32,
pub max_grad_norm: Option<f32>,
pub adaptive_step_size: bool,
pub warmup_steps: usize,
}Expand description
Configuration for aMacP optimizer
Fields§
§learning_rate: f32Base learning rate
beta1: f32First momentum coefficient (gradient averaging)
beta2: f32Second momentum coefficient (squared gradient averaging)
gamma: f32Consecutive parameter averaging coefficient
alpha: f32Dual momentum weighting factor
eta: f32Gradient heterogeneity adaptation strength
epsilon: f32Small constant for numerical stability
weight_decay: f32Weight decay coefficient
max_grad_norm: Option<f32>Maximum gradient norm for clipping
adaptive_step_size: boolEnable adaptive step size based on parameter trends
warmup_steps: usizeWarmup steps for gradient stabilization
Implementations§
Source§impl AMacPConfig
impl AMacPConfig
Sourcepub fn for_transformers() -> Self
pub fn for_transformers() -> Self
Configuration optimized for transformer models
Sourcepub fn for_vision() -> Self
pub fn for_vision() -> Self
Configuration for vision models (CNN architectures)
Sourcepub fn for_large_language_models() -> Self
pub fn for_large_language_models() -> Self
Configuration for large language models
Trait Implementations§
Source§impl Clone for AMacPConfig
impl Clone for AMacPConfig
Source§fn clone(&self) -> AMacPConfig
fn clone(&self) -> AMacPConfig
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 AMacPConfig
impl Debug for AMacPConfig
Source§impl Default for AMacPConfig
impl Default for AMacPConfig
Source§impl<'de> Deserialize<'de> for AMacPConfig
impl<'de> Deserialize<'de> for AMacPConfig
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 AMacPConfig
impl RefUnwindSafe for AMacPConfig
impl Send for AMacPConfig
impl Sync for AMacPConfig
impl Unpin for AMacPConfig
impl UnwindSafe for AMacPConfig
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