Struct DistillationConfig 

pub struct DistillationConfig {

    pub loss: DistillationLoss,
    pub temperature: Temperature,
    pub epochs: usize,
    pub learning_rate: f32,
    pub batch_size: usize,
    pub alpha: f32,
    pub optimizer: OptimizerType,
    pub lr_schedule: LearningRateSchedule,
    pub early_stopping: Option<EarlyStopping>,
    pub gradient_clip: Option<f32>,
    pub validation_split: f32,
    pub num_classes: usize,
    pub seed: u64,
}

Knowledge distillation configuration

Fields

loss: DistillationLoss

Distillation loss function

temperature: Temperature

Temperature for softening

epochs: usize

Number of training epochs

learning_rate: f32

Learning rate

batch_size: usize

Batch size

alpha: f32

Alpha weight for distillation loss (1 - alpha for hard label loss)

optimizer: OptimizerType

Optimizer type

lr_schedule: LearningRateSchedule

Learning rate schedule

early_stopping: Option<EarlyStopping>

Early stopping configuration

gradient_clip: Option<f32>

Gradient clipping threshold (None = no clipping)

validation_split: f32

Validation split ratio (0.0 to 0.3)

num_classes: usize

Number of classes for classification

seed: u64

Random seed for reproducibility

Implementations

impl DistillationConfig

pub fn builder() -> DistillationConfigBuilder

Creates a configuration builder

pub fn validate(&self) -> Result<()>

Validates the configuration

Trait Implementations

impl Clone for DistillationConfig

fn clone(&self) -> DistillationConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DistillationConfig

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

Formats the value using the given formatter.

impl Default for DistillationConfig

fn default() -> Self

Returns the “default value” for a type.