pub struct TrainingConfig {
pub default_compression_ratio: f32,
pub seed: u64,
pub gradient_clipping: Option<f32>,
}Expand description
Configuration for training acceleration.
Fields§
§default_compression_ratio: f32Default compression ratio for gradient compression.
seed: u64Random seed for reproducibility.
gradient_clipping: Option<f32>Enable gradient clipping.
Implementations§
Source§impl TrainingConfig
impl TrainingConfig
Sourcepub fn with_compression_ratio(self, ratio: f32) -> TrainingConfig
pub fn with_compression_ratio(self, ratio: f32) -> TrainingConfig
Set compression ratio.
Sourcepub fn with_seed(self, seed: u64) -> TrainingConfig
pub fn with_seed(self, seed: u64) -> TrainingConfig
Set random seed.
Sourcepub fn with_gradient_clipping(self, max_norm: f32) -> TrainingConfig
pub fn with_gradient_clipping(self, max_norm: f32) -> TrainingConfig
Enable gradient clipping.
Trait Implementations§
Source§impl Clone for TrainingConfig
impl Clone for TrainingConfig
Source§fn clone(&self) -> TrainingConfig
fn clone(&self) -> TrainingConfig
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 TrainingConfig
impl Debug for TrainingConfig
Source§impl Default for TrainingConfig
impl Default for TrainingConfig
Source§fn default() -> TrainingConfig
fn default() -> TrainingConfig
Returns the “default value” for a type. Read more
Auto Trait Implementations§
impl Freeze for TrainingConfig
impl RefUnwindSafe for TrainingConfig
impl Send for TrainingConfig
impl Sync for TrainingConfig
impl Unpin for TrainingConfig
impl UnwindSafe for TrainingConfig
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> 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