MobileV2 Pruning

This repository aim to try out different pruning-approaches on lightweight Backbones.

Usage

1. Training

   python main.py --arch MobileNetV2 (for l1norm pruner )
   python main.py --sr --arch MobileNetV2 (for slimming pruner) 
   python main.py --arch USMobileNetV2 (for Autoslim pruner )

2. Pruning (prune+finetune)

   python prune.py --arch MobileNetV2 --pruner l1normpruner --pruneratio 0.6
   python prune.py --arch MobileNetV2 --pruner SlimmingPruner --sr --pruneratio 0.6
   python prune.py --arch USMobileNetV2 --pruner AutoSlimPruner

Results on Cifar10

TODO

Pruning Methodsd

•  l1-norm pruner

•  Slimming pruner

•  AutoSlim

•  ThiNet

•  Soft filter pruning
  ....

Backbones

•  MobileV2

•  ShuffleNet
  ....

Reference

rethinking-network-pruning

Pruned-MobileNet_v2