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

[xhchrn]

trafficstars

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.