Struct PruningConfig 

pub struct PruningConfig {
    pub target_sparsity: f64,
    pub pattern: SparsityPattern,
    pub rescale: bool,
}

Configuration for the pruning process.

Fields

target_sparsity: f64

Target sparsity ratio [0, 1).

pattern: SparsityPattern

Sparsity pattern to apply.

rescale: bool

Whether to rescale remaining weights after pruning.

Implementations

impl PruningConfig

pub fn new( target_sparsity: f64, pattern: SparsityPattern, ) -> Result<Self, PruningError>

Create a new pruning config with the given sparsity and pattern.

Returns an error if target_sparsity is not in [0, 1).

pub fn with_rescale(self, rescale: bool) -> Self

Set whether to rescale non-zero weights after pruning.

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.