Struct TrainingConfig 

pub struct TrainingConfig {
    pub epochs: usize,
    pub batch_size: usize,
    pub learning_rate: f32,
    pub gradient_clip_norm: Option<f32>,
    pub gradient_accumulation_steps: usize,
    pub log_every: usize,
    pub eval_every: usize,
    pub save_checkpoints: bool,
    pub checkpoint_dir: String,
    pub mixed_precision: bool,
    pub seed: Option<u64>,
}

Configuration for training.

Fields

epochs: usize

Number of training epochs

batch_size: usize

Batch size

learning_rate: f32

Learning rate

gradient_clip_norm: Option<f32>

Gradient clipping max norm (None = no clipping)

gradient_accumulation_steps: usize

Number of gradient accumulation steps

log_every: usize

Logging frequency (steps)

eval_every: usize

Evaluation frequency (epochs)

save_checkpoints: bool

Save checkpoints

checkpoint_dir: String

Checkpoint directory

mixed_precision: bool

Use mixed precision training

seed: Option<u64>

Seed for reproducibility

Implementations

impl TrainingConfig

pub fn new() -> Self

Creates a new training configuration with defaults.

pub fn epochs(self, epochs: usize) -> Self

Builder: set number of epochs.

pub fn batch_size(self, batch_size: usize) -> Self

Builder: set batch size.

pub fn learning_rate(self, lr: f32) -> Self

Builder: set learning rate.

pub fn gradient_clip_norm(self, max_norm: f32) -> Self

Builder: set gradient clipping.

pub fn gradient_accumulation_steps(self, steps: usize) -> Self

Builder: set gradient accumulation steps.

pub fn log_every(self, steps: usize) -> Self

Builder: set logging frequency.

pub fn mixed_precision(self, enabled: bool) -> Self

Builder: enable mixed precision.

pub fn seed(self, seed: u64) -> Self

Builder: set seed.

Trait Implementations

impl Clone for TrainingConfig

fn clone(&self) -> TrainingConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for TrainingConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for TrainingConfig

fn default() -> Self

Returns the “default value” for a type.