olegpolivin / pruningExperiments
Repository to perform simple pruning experiments on neural networks
☆18Updated 4 years ago
Alternatives and similar repositories for pruningExperiments:
Users that are interested in pruningExperiments are comparing it to the libraries listed below
- ☆225Updated 8 months ago
- Contains code for the paper "Vision Transformers are Robust Learners" (AAAI 2022).☆126Updated 2 years ago
- ☆189Updated 4 years ago
- PyTorch Pruning Example☆49Updated 2 years ago
- A research library for pytorch-based neural network pruning, compression, and more.☆160Updated 2 years ago
- This repository contains a Pytorch implementation of the paper "The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networks"…☆329Updated last year
- A Pytorch Knowledge Distillation library for benchmarking and extending works in the domains of Knowledge Distillation, Pruning, and Quan…☆621Updated 2 years ago
- MUXConv: Information Multiplexing in Convolutional Neural Networks (CVPR '20), Pytorch Implementation☆51Updated 3 years ago
- ☆144Updated 2 years ago
- Repository to track the progress in model compression and acceleration☆105Updated 3 years ago
- FedCV: An Industrial-grade Federated Learning Framework for Diverse Computer Vision Tasks☆66Updated 9 months ago
- On-the-fly Structured Pruning for PyTorch models. This library implements several attributions metrics and structured pruning utils for n…☆164Updated 4 years ago
- Pytorch implementation of the paper "SNIP: Single-shot Network Pruning based on Connection Sensitivity" by Lee et al.☆107Updated 5 years ago
- Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.☆56Updated 2 years ago
- Soft Threshold Weight Reparameterization for Learnable Sparsity☆88Updated 2 years ago
- tiny-imagenet dataset downloader & reader using tensorflow_datasets (tfds) api☆20Updated 5 years ago
- DropNet: Reducing Neural Network Complexity via Iterative Pruning (ICML 2020)☆15Updated 4 years ago
- [NeurIPS'21] "AugMax: Adversarial Composition of Random Augmentations for Robust Training" by Haotao Wang, Chaowei Xiao, Jean Kossaifi, Z…☆125Updated 3 years ago
- End-to-end training of sparse deep neural networks with little-to-no performance loss.☆320Updated 2 years ago
- [NeurIPS'20] GradAug: A New Regularization Method for Deep Neural Networks☆93Updated 4 years ago
- Neural Architecture Search (NAS) papers with code☆158Updated 3 years ago
- Implementation of Continuous Sparsification, a method for pruning and ticket search in deep networks☆33Updated 2 years ago
- This repository is the official implementation of the paper Pruning via Iterative Ranking of Sensitivity Statistics and implements novel …☆32Updated last year
- Knowledge distillation from Ensembles of Iterative pruning (BMVC 2020)☆25Updated 4 years ago
- Using ideas from product quantization for state-of-the-art neural network compression.☆146Updated 3 years ago
- Implementation of Online Label Smoothing in PyTorch☆94Updated 2 years ago
- Code for "Supermasks in Superposition"☆121Updated last year
- TF2 implementation of knowledge distillation using the "function matching" hypothesis from https://arxiv.org/abs/2106.05237.☆87Updated 3 years ago
- ZSKD with PyTorch☆30Updated last year
- Neuron Merging: Compensating for Pruned Neurons (NeurIPS 2020)☆43Updated 4 years ago