On-the-fly Structured Pruning for PyTorch models. This library implements several attributions metrics and structured pruning utils for neural networks in PyTorch.
☆167Jun 4, 2020Updated 5 years ago
Alternatives and similar repositories for TorchPruner
Users that are interested in TorchPruner are comparing it to the libraries listed below
Sorting:
- Simple python code to prune pytorch model☆22Feb 11, 2020Updated 6 years ago
- [CVPR 2023] DepGraph: Towards Any Structural Pruning; LLMs, Vision Foundation Models, etc.☆3,262Sep 7, 2025Updated 5 months ago
- PyTorch implementation of "Pruning Filters For Efficient ConvNets"☆155Jul 24, 2023Updated 2 years ago
- A reproduction of PRUNING FILTERS FOR EFFICIENT CONVNETS☆28Jun 15, 2020Updated 5 years ago
- A research library for pytorch-based neural network pruning, compression, and more.☆162Nov 28, 2022Updated 3 years ago
- Repo for our Paper: Cross Quality LFW: A database for Analyzing Cross-Resolution Image Face Recognition in Unconstrained Environments☆19Nov 25, 2022Updated 3 years ago
- 针对pytorch模型的自动化模型结构分析和修改工具集，包含自动分析模型结构的模型压缩算法库☆256Apr 19, 2023Updated 2 years ago
- ☆14Mar 5, 2021Updated 4 years ago
- ResRep: Lossless CNN Pruning via Decoupling Remembering and Forgetting (ICCV 2021)☆300Dec 1, 2022Updated 3 years ago
- ☆69Mar 4, 2020Updated 6 years ago
- Learning both Weights and Connections for Efficient Neural Networks https://arxiv.org/abs/1506.02626☆181Nov 10, 2022Updated 3 years ago
- Repository to track the progress in model compression and acceleration☆106Jun 19, 2021Updated 4 years ago
- pytorch AutoSlim tools，支持三行代码对pytorch模型进行剪枝压缩☆39Apr 23, 2021Updated 4 years ago
- Curated content for DNN approximation, acceleration ... with a focus on hardware accelerator and deployment☆27May 15, 2024Updated last year
- Karras et al. (2022) diffusion models for PyTorch☆12Aug 23, 2022Updated 3 years ago
- RetinaNet face detection model inference on edge/mobile device utilizing MNN framework.☆13Feb 27, 2023Updated 3 years ago
- HWFI: Hybrid Warping Fusion for Video Frame Interpolation. IJCV 2022☆11Sep 7, 2022Updated 3 years ago
- Code and pruned models for our paper: K. Gkrispanis, N. Gkalelis, V. Mezaris, "Filter-Pruning of Lightweight Face Detectors Using a Geome…☆14May 8, 2024Updated last year
- A Pytorch implementation of Neural Network Compression (pruning, deep compression, channel pruning)☆158Jul 10, 2024Updated last year
- PyTorch library to facilitate development and standardized evaluation of neural network pruning methods.☆434Jul 7, 2023Updated 2 years ago
- [CVPR 2022] "Quarantine: Sparsity Can Uncover the Trojan Attack Trigger for Free" by Tianlong Chen*, Zhenyu Zhang*, Yihua Zhang*, Shiyu C…☆27Oct 5, 2022Updated 3 years ago
- Pruning CNN using CNN with toy example☆22Jun 21, 2021Updated 4 years ago
- quantize aware training package for NCNN on pytorch☆68Jul 27, 2021Updated 4 years ago
- The official repository for our paper "Are Neural Nets Modular? Inspecting Functional Modularity Through Differentiable Weight Masks". We…☆46Oct 3, 2023Updated 2 years ago
- A curated list of neural network pruning resources.☆2,492Apr 4, 2024Updated last year
- ☆228Jul 25, 2024Updated last year
- OpenMMLab Model Compression Toolbox and Benchmark.☆1,662Jun 11, 2024Updated last year
- We have implemented a framework that supports developers to structured prune neural networks of Tensorflow Models☆28Nov 7, 2024Updated last year
- Code for "Neuron Shapley: Discovering the Responsible Neurons"☆27May 1, 2024Updated last year
- Code for "Picking Winning Tickets Before Training by Preserving Gradient Flow" https://openreview.net/pdf?id=SkgsACVKPH☆105Feb 18, 2020Updated 6 years ago
- PyTorch Model Compression☆234Jan 27, 2023Updated 3 years ago
- Learning Efficient Convolutional Networks through Network Slimming, In ICCV 2017.☆575Jul 14, 2019Updated 6 years ago
- Prospect Pruning: Finding Trainable Weights at Initialization Using Meta-Gradients☆32Mar 30, 2022Updated 3 years ago
- Codes for DATA: Differentiable ArchiTecture Approximation.☆11Jul 22, 2021Updated 4 years ago
- Codes for reproducing the experimental results in "Proper Network Interpretability Helps Adversarial Robustness in Classification", publi…☆13Jun 29, 2020Updated 5 years ago
- Implementation of knapsack pruning☆27May 26, 2020Updated 5 years ago
- This repository is the official implementation of the paper Pruning via Iterative Ranking of Sensitivity Statistics and implements novel …☆32May 22, 2023Updated 2 years ago
- Soft Filter Pruning for Accelerating Deep Convolutional Neural Networks☆385Oct 2, 2019Updated 6 years ago
- Few training heuristics and small architectural changes that can significantly improve YOLOv3 performance with tiny increase in inference…☆12May 10, 2020Updated 5 years ago