Expand description
§Pruning
Weight-pruning strategies for model compression.
| Module | Strategy |
|---|---|
mask | SparseMask — boolean weight mask primitives |
magnitude | MagnitudePruner — unstructured L1/L2 pruning |
structured | StructuredPruner — channel / filter / head pruning |
Re-exports§
pub use magnitude::MagnitudeNorm;pub use magnitude::MagnitudePruner;pub use mask::SparseMask;pub use structured::PruneGranularity;pub use structured::StructuredPruner;
Modules§
- magnitude
- Magnitude-Based Unstructured Pruning
- mask
- Sparse Mask
- structured
- Structured Pruning