Struct MagnitudePruner 

pub struct MagnitudePruner { /* private fields */ }

Simple magnitude-based pruner.

Uses weight magnitude as importance and generates masks to achieve the target sparsity by pruning the smallest weights.

Implementations

impl MagnitudePruner

pub fn new() -> Self

Create a new magnitude pruner with L2 norm.

pub fn l1() -> Self

Create a magnitude pruner with L1 norm.

pub fn l2() -> Self

Create a magnitude pruner with L2 norm.

Trait Implementations

impl Clone for MagnitudePruner

fn clone(&self) -> MagnitudePruner

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MagnitudePruner

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

Formats the value using the given formatter.

impl Default for MagnitudePruner

fn default() -> Self

Returns the “default value” for a type.

impl Pruner for MagnitudePruner

fn generate_mask( &self, scores: &ImportanceScores, target_sparsity: f32, pattern: SparsityPattern, ) -> Result<SparsityMask, PruningError>

Generate a sparsity mask based on importance scores.

fn apply_mask( &self, module: &mut dyn Module, mask: &SparsityMask, ) -> Result<PruningResult, PruningError>

Apply a sparsity mask to a module, zeroing pruned weights.

fn importance(&self) -> &dyn Importance

Get the importance estimator used by this pruner.

fn name(&self) -> &'static str

Name of this pruner for logging.