pub struct PruningConfig {
pub sparsity: f32,
pub metric: ImportanceMetric,
pub granularity: PruningGranularity,
pub min_nonzero: usize,
}Expand description
Configuration for a pruning pass.
Fields§
§sparsity: f32Target fraction of zeros (0.0 - 1.0).
metric: ImportanceMetricWhich metric to use for computing importance scores.
granularity: PruningGranularityWhether to prune individual weights or entire rows/columns.
min_nonzero: usizeMinimum number of non-zero elements to keep (safety floor).
Implementations§
Source§impl PruningConfig
impl PruningConfig
Sourcepub fn new(
sparsity: f32,
metric: ImportanceMetric,
granularity: PruningGranularity,
) -> Self
pub fn new( sparsity: f32, metric: ImportanceMetric, granularity: PruningGranularity, ) -> Self
Create a new pruning config with the given parameters.
Sourcepub fn unstructured_l1(sparsity: f32) -> Self
pub fn unstructured_l1(sparsity: f32) -> Self
Convenience: unstructured L1-magnitude pruning at the given sparsity.
Sourcepub fn structured_row_l2(sparsity: f32) -> Self
pub fn structured_row_l2(sparsity: f32) -> Self
Convenience: structured row pruning using L2 norm at the given sparsity.
Trait Implementations§
Source§impl Clone for PruningConfig
impl Clone for PruningConfig
Source§fn clone(&self) -> PruningConfig
fn clone(&self) -> PruningConfig
Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreAuto Trait Implementations§
impl Freeze for PruningConfig
impl RefUnwindSafe for PruningConfig
impl Send for PruningConfig
impl Sync for PruningConfig
impl Unpin for PruningConfig
impl UnsafeUnpin for PruningConfig
impl UnwindSafe for PruningConfig
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> Instrument for T
impl<T> Instrument for T
Source§fn instrument(self, span: Span) -> Instrumented<Self>
fn instrument(self, span: Span) -> Instrumented<Self>
Source§fn in_current_span(self) -> Instrumented<Self>
fn in_current_span(self) -> Instrumented<Self>
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self>
fn into_either(self, into_left: bool) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more