Struct DistillationConfigBuilder 

pub struct DistillationConfigBuilder { /* private fields */ }

Builder for distillation configuration

Implementations

impl DistillationConfigBuilder

pub fn loss(self, loss: DistillationLoss) -> Self

Sets the distillation loss

pub fn temperature(self, temp: f32) -> Self

Sets the temperature

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

Sets the number of epochs

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

Sets the learning rate

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

Sets the batch size

pub fn alpha(self, alpha: f32) -> Self

Sets the alpha weight for distillation loss

pub fn optimizer(self, optimizer: OptimizerType) -> Self

Sets the optimizer type

pub fn lr_schedule(self, schedule: LearningRateSchedule) -> Self

Sets the learning rate schedule

pub fn early_stopping(self, early_stopping: Option<EarlyStopping>) -> Self

Sets early stopping configuration

pub fn gradient_clip(self, clip: Option<f32>) -> Self

Sets gradient clipping threshold

pub fn validation_split(self, split: f32) -> Self

Sets validation split ratio

pub fn num_classes(self, num: usize) -> Self

Sets number of classes

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

Sets random seed

pub fn build(self) -> DistillationConfig

Builds the configuration

Trait Implementations

impl Debug for DistillationConfigBuilder

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

Formats the value using the given formatter.

impl Default for DistillationConfigBuilder

fn default() -> DistillationConfigBuilder

Returns the “default value” for a type.