rahulvigneswaran / Lottery-Ticket-Hypothesis-in-Pytorch
This repository contains a Pytorch implementation of the paper "The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networks" by Jonathan Frankle and Michael Carbin that can be easily adapted to any model/dataset.
☆327Updated last year
Alternatives and similar repositories for Lottery-Ticket-Hypothesis-in-Pytorch:
Users that are interested in Lottery-Ticket-Hypothesis-in-Pytorch are comparing it to the libraries listed below
- ☆222Updated 6 months ago
- A repository in preparation for open-sourcing lottery ticket hypothesis code.☆627Updated 2 years ago
- ☆187Updated 4 years ago
- ☆143Updated last year
- PyHessian is a Pytorch library for second-order based analysis and training of Neural Networks☆719Updated 10 months ago
- Pytorch implementation of the paper "SNIP: Single-shot Network Pruning based on Connection Sensitivity" by Lee et al.☆106Updated 5 years ago
- PyTorch library to facilitate development and standardized evaluation of neural network pruning methods.☆428Updated last year
- A reimplementation of "The Lottery Ticket Hypothesis" (Frankle and Carbin) on MNIST.☆717Updated 4 years ago
- SNIP: SINGLE-SHOT NETWORK PRUNING BASED ON CONNECTION SENSITIVITY☆112Updated 5 years ago
- Official PyTorch implementation of Dreaming to Distill: Data-free Knowledge Transfer via DeepInversion (CVPR 2020)☆493Updated 2 years ago
- pytorch-tiny-imagenet☆170Updated last year
- Mode Connectivity and Fast Geometric Ensembles in PyTorch☆267Updated 2 years ago
- Soft Threshold Weight Reparameterization for Learnable Sparsity☆87Updated 2 years ago
- Efficient PyTorch Hessian eigendecomposition tools!☆370Updated 11 months ago
- Approximating neural network loss landscapes in low-dimensional parameter subspaces for PyTorch☆319Updated last year
- Code for "Picking Winning Tickets Before Training by Preserving Gradient Flow" https://openreview.net/pdf?id=SkgsACVKPH☆101Updated 5 years ago
- Pretrained models on CIFAR10/100 in PyTorch☆336Updated last year
- Reproduce CKA: Similarity of Neural Network Representations Revisited☆297Updated 4 years ago
- This repository contains code to replicate the experiments given in NeurIPS 2019 paper "One ticket to win them all: generalizing lottery …☆51Updated 6 months ago
- Sparse learning library and sparse momentum resources.☆379Updated 2 years ago
- End-to-end training of sparse deep neural networks with little-to-no performance loss.☆317Updated 2 years ago
- Code for Neural Architecture Search without Training (ICML 2021)☆464Updated 3 years ago
- Learning Sparse Neural Networks through L0 regularization☆239Updated 4 years ago
- A research library for pytorch-based neural network pruning, compression, and more.☆160Updated 2 years ago
- Pruning Neural Networks with Taylor criterion in Pytorch☆315Updated 5 years ago
- A simple code snippet for freezing only part of the layer☆33Updated 3 years ago
- Zero-Cost Proxies for Lightweight NAS☆147Updated last year
- Code release for "Adversarial Robustness vs Model Compression, or Both?"☆91Updated 3 years ago
- ☆156Updated 2 years ago
- [ICLR 2021] "Neural Architecture Search on ImageNet in Four GPU Hours: A Theoretically Inspired Perspective" by Wuyang Chen, Xinyu Gong, …☆167Updated 3 years ago