• 设为首页
  • 点击收藏
  • 手机版
    手机扫一扫访问
    迪恩网络手机版
  • 关注官方公众号
    微信扫一扫关注
    迪恩网络公众号

apxlwl/MobileNet-v2-pruning: Try out different pruning-approaches on lightweight ...

原作者: [db:作者] 来自: 网络 收藏 邀请

开源软件名称(OpenSource Name):

apxlwl/MobileNet-v2-pruning

开源软件地址(OpenSource Url):

https://github.com/apxlwl/MobileNet-v2-pruning

开源编程语言(OpenSource Language):

Python 100.0%

开源软件介绍(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

BackBone Pruner Prune Ratio Original/Pruned/Finetuned Accuracy FLOPs(M) Params(M)
MobileV2 L1-Norm 0.6 0.937/0.100/0.844 313.5->225.5 2.24->1.15
MobileV2 Slimming Optimal Thres 0.922/0.485/0.915 313.5->127.5 2.24->0.98
MobileV2 AutoSlim <200 flops 0.922/0.795/0.919 313.5->137.5 2.24->1.037
VGG Slimming Optimal Thres 0.926/0.183/0.920 399.3->147.8 20.03->1.49
Resnet50 Slimming Optimal Thres 0.926/0.665/0.921 3448->975 23.52->6.00
ShuffleNetV2 Slimming Optimal Thres 0.897/0.894/0.895 348.6->188.5 2.22->1.24

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

Backbones

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

Reference

rethinking-network-pruning

Pruned-MobileNet_v2




鲜花

握手

雷人

路过

鸡蛋
该文章已有0人参与评论

请发表评论

全部评论

专题导读
热门推荐
阅读排行榜

扫描微信二维码

查看手机版网站

随时了解更新最新资讯

139-2527-9053

在线客服(服务时间 9:00~18:00)

在线QQ客服
地址:深圳市南山区西丽大学城创智工业园
电邮:jeky_zhao#qq.com
移动电话:139-2527-9053

Powered by 互联科技 X3.4© 2001-2213 极客世界.|Sitemap