lucaslie / torchprune
A research library for pytorch-based neural network pruning, compression, and more.
☆160Updated last year
Related projects ⓘ
Alternatives and complementary repositories for torchprune
- Soft Threshold Weight Reparameterization for Learnable Sparsity☆88Updated last year
- ☆219Updated 3 months ago
- PyTorch library to facilitate development and standardized evaluation of neural network pruning methods.☆424Updated last year
- Code for "Picking Winning Tickets Before Training by Preserving Gradient Flow" https://openreview.net/pdf?id=SkgsACVKPH☆101Updated 4 years ago
- [ICLR 2020] Drawing Early-Bird Tickets: Toward More Efficient Training of Deep Networks☆137Updated 4 years ago
- Low Precision Arithmetic Simulation in PyTorch☆265Updated 6 months ago
- [CVPR'20] ZeroQ: A Novel Zero Shot Quantization Framework☆274Updated 11 months ago
- [ICLR 2021] "Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired Perspective" by Wuyang Chen, Xinyu Gong, …☆167Updated 2 years ago
- [ICLR 2021] HW-NAS-Bench: Hardware-Aware Neural Architecture Search Benchmark☆104Updated last year
- ☆194Updated last year
- code for "AttentiveNAS Improving Neural Architecture Search via Attentive Sampling"☆103Updated 3 years ago
- ☆42Updated 9 months ago
- Identify a binary weight or binary weight and activation subnetwork within a randomly initialized network by only pruning and binarizing …☆49Updated 2 years ago
- Pruning Neural Networks with Taylor criterion in Pytorch☆314Updated 5 years ago
- Model compression by constrained optimization, using the Learning-Compression (LC) algorithm☆69Updated 2 years ago
- ☆70Updated 4 years ago
- Pytorch implementation of the paper "SNIP: Single-shot Network Pruning based on Connection Sensitivity" by Lee et al.☆105Updated 5 years ago
- [ICLR 2021] "CPT: Efficient Deep Neural Network Training via Cyclic Precision" by Yonggan Fu, Han Guo, Meng Li, Xin Yang, Yining Ding, Vi…☆30Updated 8 months ago
- Code accompanying the NeurIPS 2020 paper: WoodFisher (Singh & Alistarh, 2020)☆46Updated 3 years ago
- ☆186Updated 3 years ago
- A general and accurate MACs / FLOPs profiler for PyTorch models☆571Updated 6 months ago
- Implementation for the paper "Latent Weights Do Not Exist: Rethinking Binarized Neural Network Optimization"☆73Updated 4 years ago
- End-to-end training of sparse deep neural networks with little-to-no performance loss.☆317Updated last year
- Implementation of Continuous Sparsification, a method for pruning and ticket search in deep networks☆32Updated 2 years ago
- Using ideas from product quantization for state-of-the-art neural network compression.☆145Updated 3 years ago
- Quantization library for PyTorch. Support low-precision and mixed-precision quantization, with hardware implementation through TVM.☆413Updated last year
- Binarize convolutional neural networks using pytorch☆134Updated 2 years ago
- ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training☆201Updated last year
- A library for researching neural networks compression and acceleration methods.☆136Updated 2 months ago
- [ICLR 2022] The Unreasonable Effectiveness of Random Pruning: Return of the Most Naive Baseline for Sparse Training by Shiwei Liu, Tianlo…☆73Updated last year