/pytorch_pruning

Implementation of a few NN pruning methods in PyTorch, alongside testing of the Lottery Ticket Hypothesis

Primary LanguageJupyter Notebook

Stargazers