StijnVerdenius / SNIP-itLinks
This repository is the official implementation of the paper Pruning via Iterative Ranking of Sensitivity Statistics and implements novel pruning / compression algorithms for deep learning / neural networks. Amongst others it implements structured pruning before training, its actual parameter shrinking and unstructured before/during training.
☆32Updated 2 years ago
Alternatives and similar repositories for SNIP-it
Users that are interested in SNIP-it are comparing it to the libraries listed below
Sorting:
- ☆14Updated 4 years ago
- Lookahead: A Far-sighted Alternative of Magnitude-based Pruning (ICLR 2020)☆32Updated 5 years ago
- Code for "Picking Winning Tickets Before Training by Preserving Gradient Flow" https://openreview.net/pdf?id=SkgsACVKPH☆105Updated 5 years ago
- Soft Threshold Weight Reparameterization for Learnable Sparsity☆91Updated 2 years ago
- ☆228Updated last year
- ☆194Updated this week
- Estimate/count FLOPS for a given neural network using pytorch☆305Updated 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
- SNIP: SINGLE-SHOT NETWORK PRUNING BASED ON CONNECTION SENSITIVITY☆116Updated 6 years ago
- Differentiable Data Augmentation Library☆125Updated 3 years ago
- A research library for pytorch-based neural network pruning, compression, and more.☆162Updated 3 years ago
- A Signal Propagation Perspective for Pruning Neural Networks at Initialization☆14Updated 5 years ago
- Code for paper "Orthogonal Convolutional Neural Networks".☆118Updated 4 years ago
- Train ImageNet *fast* in 500 lines of code with FFCV☆149Updated last year
- A PyTorch converter for SimCLR checkpoints☆108Updated 5 years ago
- PyTorch implementation of "Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning" with DDP and Apex AMP☆84Updated 5 years ago
- PyTorch implementation of shake-drop regularization☆55Updated 5 years ago
- MUXConv: Information Multiplexing in Convolutional Neural Networks (CVPR '20), Pytorch Implementation☆52Updated 4 years ago
- Is the attention layer even necessary? (https://arxiv.org/abs/2105.02723)☆485Updated 4 years ago
- Codebase for the paper "A Gradient Flow Framework for Analyzing Network Pruning"☆20Updated 5 years ago
- Neural Architecture Transfer (Arxiv'20), PyTorch Implementation☆155Updated 5 years ago
- [ICLR 2021] "Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired Perspective" by Wuyang Chen, Xinyu Gong, …☆168Updated 4 years ago
- [ICLR 2020] Drawing Early-Bird Tickets: Toward More Efficient Training of Deep Networks☆140Updated 5 years ago
- DeepHoyer: Learning Sparser Neural Network with Differentiable Scale-Invariant Sparsity Measures☆32Updated 5 years ago
- Code for our ICLR'2021 paper "DrNAS: Dirichlet Neural Architecture Search"☆43Updated 4 years ago
- Seamless analysis of your PyTorch models (RAM usage, FLOPs, MACs, receptive field, etc.)☆224Updated 10 months ago
- code for "AttentiveNAS Improving Neural Architecture Search via Attentive Sampling"☆105Updated 4 years ago
- ☆13Updated 4 years ago
- [ICLR 2020] NAS evaluation is frustratingly hard☆149Updated 2 years ago
- A new test set for ImageNet☆262Updated 2 years ago