pub struct ProdigyParameterState {
pub momentum: Vec<f32>,
pub variance: Vec<f32>,
pub distance: f64,
pub step: usize,
}Expand description
Optimizer state for individual parameters.
Fields§
§momentum: Vec<f32>First moment estimate (momentum)
variance: Vec<f32>Second moment estimate (variance)
distance: f64Current distance estimate
step: usizeStep count for this parameter
Implementations§
Source§impl ProdigyParameterState
impl ProdigyParameterState
pub fn new(param_size: usize, initial_distance: f64) -> Self
Sourcepub fn memory_usage(&self) -> ProdigyMemoryStats
pub fn memory_usage(&self) -> ProdigyMemoryStats
Get memory usage statistics for this parameter state.
Trait Implementations§
Source§impl Clone for ProdigyParameterState
impl Clone for ProdigyParameterState
Source§fn clone(&self) -> ProdigyParameterState
fn clone(&self) -> ProdigyParameterState
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 ProdigyParameterState
impl Debug for ProdigyParameterState
Source§impl<'de> Deserialize<'de> for ProdigyParameterState
impl<'de> Deserialize<'de> for ProdigyParameterState
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 ProdigyParameterState
impl RefUnwindSafe for ProdigyParameterState
impl Send for ProdigyParameterState
impl Sync for ProdigyParameterState
impl Unpin for ProdigyParameterState
impl UnwindSafe for ProdigyParameterState
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