开源软件名称（OpenSource Name）：

开源软件地址(OpenSource Url)：

开源编程语言(OpenSource Language)：

开源软件介绍(OpenSource Introduction)：

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

NOTE:

1. args for VGG: --arch VGG --s 0.001 --sr --lr 0.02 --epochs 100
2. args for resnet: --arch resnet50 --s 0.001 --sr --lr 0.02 --epochs 100
3. args for shufflenet: --arch ShuffleNetV2 --s 0.007 --sr --lr 0.001 --epochs 100
   Try yourself with different arguments!

TODO

Pruning Methodsd

• l1-norm pruner
• Slimming pruner
• AutoSlim
• ThiNet
• Soft filter pruning
  ....

Backbones

• MobileV2
• ShuffleNet
• VGG
• ResNet
  ....

Reference

rethinking-network-pruning

Pruned-MobileNet_v2