pub struct TrainingConfig {
pub batch_size: usize,
pub shuffle: bool,
pub num_workers: usize,
pub learning_rate: f64,
pub epochs: usize,
pub verbose: usize,
pub validation: Option<ValidationSettings>,
pub gradient_accumulation: Option<GradientAccumulationConfig>,
pub mixed_precision: Option<MixedPrecisionConfig>,
}
Expand description
Configuration structure for training neural networks
Fields§
§batch_size: usize
Number of samples in each training batch
shuffle: bool
Whether to shuffle the training data between epochs
num_workers: usize
Number of parallel workers for data loading
learning_rate: f64
Base learning rate for the optimizer
epochs: usize
Number of complete passes through the training dataset
verbose: usize
Verbosity level for training output
validation: Option<ValidationSettings>
Validation configuration
gradient_accumulation: Option<GradientAccumulationConfig>
Gradient accumulation configuration
mixed_precision: Option<MixedPrecisionConfig>
Mixed precision training configuration
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
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> 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