/lottery-ticket-hyopothesis

PyTorch implementation of The Lottery Tickets Hypothesis.

Primary LanguagePython

PyTorch implementation of [1803.03635 The Lottery Ticket Hypothesis: Finding Sparse, Trainable Neural Networks]

This demonstrates finding a Lenet based winning tickets(structure) that classifies MNIST.

Usage

python main.py --prune_type=['choose from normal, resample, reinit']