lecode-official/pytorch-lottery-ticket-hypothesis

A PyTorch implementation of the Lottery Ticket algorithm introduced by Frankle et al. in "The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networks" and enhanced by Zhou et al. in "Deconstructing Lottery Tickets: Zeros, Signs, and the Supermask".

PythonMIT

Issues

Log hyperparameters and training statistics
#2 opened 2 years ago by lecode-official
0
Save the trained neural networks and the pruning masks
#1 opened 2 years ago by lecode-official
0
Add support for training without iterative magnitude pruning
#9 opened 2 years ago by lecode-official
0
Add support for ResNet-18
#6 opened 2 years ago by lecode-official
0
Add support for plotting the training statistics
#3 opened 2 years ago by lecode-official
0
Make it possible to gracefully abort the training process
#4 opened 2 years ago by lecode-official
0
Add support for macOS on ARM64
#5 opened 2 years ago by lecode-official
0
Replicate the experiments from the original paper
#7 opened 2 years ago by lecode-official
0
Add optional dropout support
#8 opened 2 years ago by lecode-official
0