Struct PruningConfig 

pub struct PruningConfig { /* private fields */ }

Configuration for pruning operations.

Example

use entrenar::prune::{PruningConfig, PruningSchedule, PruneMethod};

let config = PruningConfig::default()
    .with_method(PruneMethod::Wanda)
    .with_schedule(PruningSchedule::Gradual {
        start_step: 1000,
        end_step: 5000,
        initial_sparsity: 0.0,
        final_sparsity: 0.5,
        frequency: 100,
    })
    .with_target_sparsity(0.5);

Implementations

impl PruningConfig

pub fn new() -> Self

Create a new configuration with default values.

pub fn with_method(self, method: PruneMethod) -> Self

Set the pruning method.

pub fn with_target_sparsity(self, sparsity: f32) -> Self

Set the target sparsity.

pub fn with_pattern(self, pattern: SparsityPatternConfig) -> Self

Set the sparsity pattern.

pub fn with_schedule(self, schedule: PruningSchedule) -> Self

Set the pruning schedule.

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

Enable or disable fine-tuning after pruning.

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

Set the number of fine-tuning steps.

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

Set the fine-tuning learning rate.

pub fn with_skip_embed_layers(self, skip: bool) -> Self

Enable or disable skipping embedding layers.

pub fn method(&self) -> PruneMethod

Get the pruning method.

pub fn target_sparsity(&self) -> f32

Get the target sparsity.

pub fn pattern(&self) -> &SparsityPatternConfig

Get the sparsity pattern.

pub fn schedule(&self) -> &PruningSchedule

Get the pruning schedule.

pub fn fine_tune_after_pruning(&self) -> bool

Check if fine-tuning is enabled.

pub fn fine_tune_steps(&self) -> usize

Get fine-tuning steps.

pub fn fine_tune_lr(&self) -> f32

Get fine-tuning learning rate.

pub fn skip_embed_layers(&self) -> bool

Check if embedding layers should be skipped.

pub fn requires_calibration(&self) -> bool

Check if this configuration requires calibration data.

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

Validate the configuration.

Trait Implementations

impl Clone for PruningConfig

fn clone(&self) -> PruningConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PruningConfig

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

Formats the value using the given formatter.

impl Default for PruningConfig

fn default() -> Self

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for PruningConfig

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for PruningConfig

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

where __S: Serializer,

Serialize this value into the given Serde serializer.