Lottery-Ticket-Hypothesis-in-Pytorch
Lottery-Ticket-Hypothesis-in-Pytorch copied to clipboard
The implementation uses layer-wise pruning, while global pruning is usually used for large conv models
The prune_by_percentile
function defined in main.py
uses layer-wise pruning for all models. While it is found in the original LTH paper that global pruning works better for larger convolutional models such as ResNet and VGG.