Enum PruningMethod

pub enum PruningMethod {
    MagnitudeBased {
        threshold: f64,
    },
    TopK {
        k: usize,
    },
    Structured {
        granularity: StructuredGranularity,
    },
    GradualMagnitude {
        initial_sparsity: f64,
        final_sparsity: f64,
        begin_step: usize,
        end_step: usize,
    },
}

Pruning method

Variants

MagnitudeBased

Magnitude-based pruning

Fields

threshold: f64

Threshold below which weights are pruned

TopK

Top-k pruning (keep only k largest weights)

Fields

k: usize

Number of largest weights to keep

Structured

Structured pruning (remove entire channels/filters)

Fields

granularity: StructuredGranularity

Granularity level for structured pruning

GradualMagnitude

Gradual magnitude pruning

Fields

initial_sparsity: f64

Initial sparsity ratio (0.0 to 1.0)

final_sparsity: f64

Final sparsity ratio (0.0 to 1.0)

begin_step: usize

Training step to begin pruning

end_step: usize

Training step to end pruning

Trait Implementations

impl Clone for PruningMethod

fn clone(&self) -> PruningMethod

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PruningMethod

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

Formats the value using the given formatter.

impl PartialEq for PruningMethod

fn eq(&self, other: &PruningMethod) -> bool

Tests for self and other values to be equal, and is used by ==.

const fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for PruningMethod