Struct PruneTrainerConfig 

pub struct PruneTrainerConfig {
    pub pruning: PruningConfig,
    pub calibration: CalibrationDataConfig,
    pub finetune_epochs: usize,
    pub finetune_lr: f32,
    pub evaluate_pre_post: bool,
    pub checkpoint_dir: Option<String>,
    pub save_checkpoints: bool,
}

Configuration for the prune-finetune trainer.

Fields

pruning: PruningConfig

Pruning configuration.

calibration: CalibrationDataConfig

Calibration data configuration.

finetune_epochs: usize

Number of fine-tuning epochs after pruning.

finetune_lr: f32

Learning rate for fine-tuning.

evaluate_pre_post: bool

Whether to evaluate before and after pruning.

checkpoint_dir: Option<String>

Checkpoint directory.

save_checkpoints: bool

Whether to save intermediate checkpoints.

Implementations

impl PruneTrainerConfig

pub fn new() -> Self

Create a new configuration with default values.

pub fn with_pruning(self, config: PruningConfig) -> Self

Set the pruning configuration.

pub fn with_calibration(self, config: CalibrationDataConfig) -> Self

Set the calibration configuration.

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

Set the number of fine-tuning epochs.

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

Set the fine-tuning learning rate.

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

Enable or disable pre/post evaluation.

pub fn with_checkpoint_dir(self, dir: impl Into<String>) -> Self

Set the checkpoint directory.

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

Enable or disable checkpoint saving.

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

Validate the configuration.

Trait Implementations

impl Clone for PruneTrainerConfig

fn clone(&self) -> PruneTrainerConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PruneTrainerConfig

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

Formats the value using the given formatter.

impl Default for PruneTrainerConfig

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for PruneTrainerConfig

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for PruneTrainerConfig

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.