← all repositories

Eric-mingjie/rethinking-network-pruning

PyTorch implementation of a research paper investigating whether structured network pruning provides benefits compared to training models from scratch.

1.5k stars Python ML FrameworksLearning
rethinking-network-pruning
Velocity · 7d
+0.5
★ / day
Trend
steady
star history

This repository reproduces the ICLR 2019 paper demonstrating that for structured network pruning, training a pruned model from scratch often achieves comparable or better accuracy than the traditional train-prune-finetune approach. The work challenges common assumptions about the value of learned “important” weights in pruning, arguing that the pruned architecture itself is more crucial to final model efficiency. The code includes implementations of several pruning methods and trained ImageNet models for reproducibility.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.