lucaslie / torchpruneLinks
A research library for pytorch-based neural network pruning, compression, and more.
☆163Updated 3 years ago
Alternatives and similar repositories for torchprune
Users that are interested in torchprune are comparing it to the libraries listed below
Sorting:
- ☆227Updated last year
- Soft Threshold Weight Reparameterization for Learnable Sparsity☆90Updated 2 years ago
- ☆43Updated last year
- Implementation for the paper "Latent Weights Do Not Exist: Rethinking Binarized Neural Network Optimization"☆75Updated 6 years ago
- PyTorch library to facilitate development and standardized evaluation of neural network pruning methods.☆433Updated 2 years ago
- Binarize convolutional neural networks using pytorch☆149Updated 3 years ago
- [ICLR 2020] Drawing Early-Bird Tickets: Toward More Efficient Training of Deep Networks☆140Updated 5 years ago
- Code accompanying the NeurIPS 2020 paper: WoodFisher (Singh & Alistarh, 2020)☆53Updated 4 years ago
- Reference implementations of popular Binarized Neural Networks☆109Updated 3 weeks ago
- Identify a binary weight or binary weight and activation subnetwork within a randomly initialized network by only pruning and binarizing …☆51Updated 3 years ago
- ☆69Updated 5 years ago
- [ICLR 2021 Spotlight] "CPT: Efficient Deep Neural Network Training via Cyclic Precision" by Yonggan Fu, Han Guo, Meng Li, Xin Yang, Yinin…☆31Updated last year
- Low Precision Arithmetic Simulation in PyTorch☆287Updated last year
- Code for "Picking Winning Tickets Before Training by Preserving Gradient Flow" https://openreview.net/pdf?id=SkgsACVKPH☆105Updated 5 years ago
- Pytorch implementation of our paper accepted by NeurIPS 2020 -- Rotated Binary Neural Network☆83Updated 2 years ago
- On-the-fly Structured Pruning for PyTorch models. This library implements several attributions metrics and structured pruning utils for n…☆167Updated 5 years ago
- End-to-end training of sparse deep neural networks with little-to-no performance loss.☆331Updated 2 years ago
- Using ideas from product quantization for state-of-the-art neural network compression.☆146Updated 4 years ago
- [ICLR 2021] "Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired Perspective" by Wuyang Chen, Xinyu Gong, …☆168Updated 3 years ago
- Pytorch implementation of the paper "SNIP: Single-shot Network Pruning based on Connection Sensitivity" by Lee et al.☆110Updated 6 years ago
- Lightweight torch implementation of rigl, a sparse-to-sparse optimizer.☆60Updated 4 years ago
- code for "AttentiveNAS Improving Neural Architecture Search via Attentive Sampling"☆105Updated 4 years ago
- [ICLR 2021] HW-NAS-Bench: Hardware-Aware Neural Architecture Search Benchmark☆113Updated 2 years ago
- ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training☆199Updated 3 years ago
- AlphaNet Improved Training of Supernet with Alpha-Divergence☆100Updated 4 years ago
- [CVPR'20] ZeroQ: A Novel Zero Shot Quantization Framework☆279Updated 2 years ago
- All about acceleration and compression of Deep Neural Networks☆33Updated 6 years ago
- Estimate/count FLOPS for a given neural network using pytorch☆305Updated 3 years ago
- Model compression by constrained optimization, using the Learning-Compression (LC) algorithm☆72Updated 4 years ago
- Official PyTorch Implementation of HELP: Hardware-adaptive Efficient Latency Prediction for NAS via Meta-Learning (NeurIPS 2021 Spotlight…☆63Updated last year