A research library for pytorch-based neural network pruning, compression, and more.
☆162Nov 28, 2022Updated 3 years ago
Alternatives and similar repositories for torchprune
Users that are interested in torchprune are comparing it to the libraries listed below
Sorting:
- Revisiting Parameter Sharing for Automatic Neural Channel Number Search, NeurIPS 2020☆22Nov 15, 2020Updated 5 years ago
- ☆22May 27, 2018Updated 7 years ago
- Official PyTorch implementation for our ICCV 2019 paper - Fooling Network Interpretation in Image Classification☆24Nov 21, 2019Updated 6 years ago
- A curated list of neural network pruning resources.☆2,492Apr 4, 2024Updated last year
- Personal Digest of NAS (Under Construction 🛠)☆25Nov 24, 2020Updated 5 years ago
- ☆43Jan 30, 2024Updated 2 years ago
- PyTorch implementation for OD-cheap-convolution.☆20Sep 29, 2019Updated 6 years ago
- (CVPR 2021, Oral) Dynamic Slimmable Network☆231Dec 31, 2021Updated 4 years ago
- Accompanying code for our paper "Optimizing Short-Time Fourier Transform Parameters via Gradient Descent"☆34Oct 30, 2020Updated 5 years ago
- Low Precision Arithmetic Simulation in PyTorch☆292May 20, 2024Updated last year
- Enable RNNLM lattice rescoring with Pytorch [kaldi]☆12Jun 5, 2020Updated 5 years ago
- Code for the paper: Unified Gradient Reweighting for Model Biasing with Applications to Source Separation☆14Nov 16, 2020Updated 5 years ago
- An implementation of the Prism layer (https://arxiv.org/abs/2011.04823)☆12Nov 13, 2020Updated 5 years ago
- Code of our Neurips2020 paper "Auto Learning Attention", coming soon☆22Apr 14, 2021Updated 4 years ago
- Code for the ICML 2021 paper "Sharing Less is More: Lifelong Learning in Deep Networks with Selective Layer Transfer"☆12Aug 17, 2021Updated 4 years ago
- ☆15Jan 8, 2020Updated 6 years ago
- [ICLR 2021] "Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired Perspective" by Wuyang Chen, Xinyu Gong, …☆169Dec 30, 2021Updated 4 years ago
- (ECCV'2020 Oral)EagleEye: Fast Sub-net Evaluation for Efficient Neural Network Pruning☆309Dec 8, 2022Updated 3 years ago
- Solos: A Dataset for Audio-Visual Music Analysis☆24Feb 17, 2023Updated 3 years ago
- ☆35Mar 4, 2020Updated 6 years ago
- ☆15Dec 7, 2021Updated 4 years ago
- ICML2019 Accepted Paper. Overcoming Multi-Model Forgetting☆14Jun 5, 2019Updated 6 years ago
- Pruning Neural Networks with Taylor criterion in Pytorch☆321Nov 3, 2019Updated 6 years ago
- Interpolation between Residual and Non-Residual Networks, ICML 2020. https://arxiv.org/abs/2006.05749☆26Aug 16, 2020Updated 5 years ago
- The code for Joint Neural Architecture Search and Quantization☆14Apr 10, 2019Updated 6 years ago
- ☆41Apr 3, 2021Updated 4 years ago
- On-the-fly Structured Pruning for PyTorch models. This library implements several attributions metrics and structured pruning utils for n…☆167Jun 4, 2020Updated 5 years ago
- Using ideas from product quantization for state-of-the-art neural network compression.☆145Aug 14, 2021Updated 4 years ago
- Code for Cross-dataset Training☆15Dec 27, 2020Updated 5 years ago
- WeightNet: Revisiting the Design Space of Weight Networks☆19Feb 20, 2021Updated 5 years ago
- Code for the ICLR2020 "Training Binary Neural Networks with Real-to-Binary Convolutions☆34Jun 16, 2020Updated 5 years ago
- ☆30Dec 28, 2021Updated 4 years ago
- Pytorch implementation of our paper accepted by IEEE TNNLS, 2022 -- Distilling a Powerful Student Model via Online Knowledge Distillation☆31Nov 11, 2021Updated 4 years ago
- Implementation of knapsack pruning☆27May 26, 2020Updated 5 years ago
- This is the official repository for Batch Level Distillation (BLD)☆15Jan 25, 2021Updated 5 years ago
- ☆17Jul 10, 2022Updated 3 years ago
- Training with Block Minifloat number representation☆18May 2, 2021Updated 4 years ago
- Implementation of Neuron-level Structured Pruning using Polarization Regularizer☆82Nov 22, 2022Updated 3 years ago
- Soft Threshold Weight Reparameterization for Learnable Sparsity☆91Feb 15, 2023Updated 3 years ago