pub struct TrainingConfig {Show 15 fields
pub learning_rate: f32,
pub batch_size: usize,
pub epochs: usize,
pub validation_split: f32,
pub early_stopping_patience: Option<usize>,
pub lr_decay: f32,
pub lr_decay_step: usize,
pub grad_clip: f32,
pub adam_beta1: f32,
pub adam_beta2: f32,
pub adam_epsilon: f32,
pub l2_reg: f32,
pub enable_distillation: bool,
pub distillation_temperature: f32,
pub distillation_alpha: f32,
}Expand description
Training hyperparameters
Fields§
§learning_rate: f32Learning rate
batch_size: usizeBatch size
epochs: usizeNumber of epochs
validation_split: f32Validation split ratio (0.0 to 1.0)
early_stopping_patience: Option<usize>Early stopping patience (epochs)
lr_decay: f32Learning rate decay factor
lr_decay_step: usizeLearning rate decay step (epochs)
grad_clip: f32Gradient clipping threshold
adam_beta1: f32Adam beta1 parameter
adam_beta2: f32Adam beta2 parameter
adam_epsilon: f32Adam epsilon for numerical stability
l2_reg: f32L2 regularization strength
enable_distillation: boolEnable knowledge distillation
distillation_temperature: f32Temperature for distillation
distillation_alpha: f32Alpha for balancing hard and soft targets (0.0 = only hard, 1.0 = only soft)
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§impl<'de> Deserialize<'de> for TrainingConfig
impl<'de> Deserialize<'de> for TrainingConfig
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 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