Skip to content

ktkth5/lottery-ticket-hyopothesis

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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']

About

PyTorch implementation of The Lottery Tickets Hypothesis.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages