Lottery-Ticket-Hypothesis-in-Pytorch icon indicating copy to clipboard operation
Lottery-Ticket-Hypothesis-in-Pytorch copied to clipboard

The implementation uses layer-wise pruning, while global pruning is usually used for large conv models

Open xhchrn opened this issue 4 years ago • 0 comments

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.

xhchrn avatar Oct 21 '20 18:10 xhchrn