Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
mobilenet_prune_pytorch
Commits
0d941848
"vscode:/vscode.git/clone" did not exist on "88f2245d86b59f11d92ff621e9fdcb697c2b37f2"
Commit
0d941848
authored
Dec 08, 2023
by
chenxj
Browse files
update README
parent
9c887bae
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
22 additions
and
8 deletions
+22
-8
README.md
README.md
+19
-7
model.properties
model.properties
+3
-1
No files found.
README.md
View file @
0d941848
# mobilenet_prune
# mobilenet_prune
## 模型介绍
## 论文
基于torch-pruning对mobilenetv2、mobilenetv3、efficientnet进行剪枝。
EfficientNet
https://arxiv.org/pdf/1905.11946.pdf
mobilenetv2
https://arxiv.org/pdf/1801.04381.pdf
mobilenetv3
https://arxiv.org/pdf/1905.02244.pdf
## 模型结构
## 模型结构
模型结构见官方论文:
模型结构见官方论文:
EfficientNet
EfficientNet
...
@@ -9,10 +14,11 @@ mobilenetv2
...
@@ -9,10 +14,11 @@ mobilenetv2
https://arxiv.org/pdf/1801.04381.pdf
https://arxiv.org/pdf/1801.04381.pdf
mobilenetv3
mobilenetv3
https://arxiv.org/pdf/1905.02244.pdf
https://arxiv.org/pdf/1905.02244.pdf
## 算法原理
CNN原理
## 数据集
## 数据集
torchvision.datasets.CIFAR10
torchvision.datasets.CIFAR10
## 剪枝及微调
## 环境配置
### 环境配置
在
[
光源
](
https://www.sourcefind.cn/#/service-details
)
可拉取剪枝及微调的docker镜像,在
[
光合开发者社区
](
https://cancon.hpccube.com:65024/4/main/
)
可下载torch安装包。mobilenet_prune推荐的镜像如下:
在
[
光源
](
https://www.sourcefind.cn/#/service-details
)
可拉取剪枝及微调的docker镜像,在
[
光合开发者社区
](
https://cancon.hpccube.com:65024/4/main/
)
可下载torch安装包。mobilenet_prune推荐的镜像如下:
```
```
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py37-latest
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04-py37-latest
...
@@ -27,7 +33,7 @@ cd mobilenet_prune
...
@@ -27,7 +33,7 @@ cd mobilenet_prune
pip3 install -r requirements.txt
pip3 install -r requirements.txt
```
```
##
#
剪枝
## 剪枝
efficientnet
efficientnet
```
```
python3 main.py --mode prune --model efficientnet --batch-size 128 --restore ./checkpoints/efficientnet.pt --dataset cifar10 --method l1 --speed-up 2.11 --global-pruning
python3 main.py --mode prune --model efficientnet --batch-size 128 --restore ./checkpoints/efficientnet.pt --dataset cifar10 --method l1 --speed-up 2.11 --global-pruning
...
@@ -40,7 +46,7 @@ mobilenetv3
...
@@ -40,7 +46,7 @@ mobilenetv3
```
```
python3 main.py --mode prune --model mobilenetv3 --batch-size 128 --restore ./checkpoints/mobilenetv3.pt --dataset cifar10 --method l1 --speed-up 2.11 --global-pruning
python3 main.py --mode prune --model mobilenetv3 --batch-size 128 --restore ./checkpoints/mobilenetv3.pt --dataset cifar10 --method l1 --speed-up 2.11 --global-pruning
```
```
##
#
测试
## 测试
efficientnet
efficientnet
```
```
python3 main.py --mode test --model efficientnet --batch-size 128 --restore ./run/cifar10/prune/cifar10-global-l1-efficientnet/cifar10_efficientnet_l1.pt --dataset cifar10 --method l1 --speed-up 2.11 --global-pruning
python3 main.py --mode test --model efficientnet --batch-size 128 --restore ./run/cifar10/prune/cifar10-global-l1-efficientnet/cifar10_efficientnet_l1.pt --dataset cifar10 --method l1 --speed-up 2.11 --global-pruning
...
@@ -53,7 +59,8 @@ mobilenetv3
...
@@ -53,7 +59,8 @@ mobilenetv3
```
```
python3 main.py --mode test --model mobilenetv3 --batch-size 128 --restore ./run/cifar10/prune/cifar10-global-l1-mobilenetv3/cifar10_mobilenetv3_l1.pt --dataset cifar10 --method l1 --speed-up 2.11 --global-pruning
python3 main.py --mode test --model mobilenetv3 --batch-size 128 --restore ./run/cifar10/prune/cifar10-global-l1-mobilenetv3/cifar10_mobilenetv3_l1.pt --dataset cifar10 --method l1 --speed-up 2.11 --global-pruning
```
```
## 性能和准确率数据
## result
### 精度
method l1
method l1
| model | Base Acc | Pruned Acc | infer time(ms)| pruned infer time(ms)|
| model | Base Acc | Pruned Acc | infer time(ms)| pruned infer time(ms)|
| :------: | :------: |:------: |:------: |:------: |
| :------: | :------: |:------: |:------: |:------: |
...
@@ -75,6 +82,11 @@ method group_norm
...
@@ -75,6 +82,11 @@ method group_norm
| mobilenetv2 | 0.9222 | 0.8953 | 0.0002316784713882953 | 0.0001845753353089094 |
| mobilenetv2 | 0.9222 | 0.8953 | 0.0002316784713882953 | 0.0001845753353089094 |
| mobilenetv3 | 0.9034 | 0.8852 | 0.0001795800439082086 | 0.00017950512771494688 |
| mobilenetv3 | 0.9034 | 0.8852 | 0.0001795800439082086 | 0.00017950512771494688 |
## 应用场景
### 算法类别
图像分类
### 热点应用行业
交通,金融,医疗,教育,家居
## 源码仓库及问题反馈
## 源码仓库及问题反馈
https://developer.hpccube.com/codes/modelzoo/mobilenet_prune
https://developer.hpccube.com/codes/modelzoo/mobilenet_prune
## 参考
## 参考
...
...
model.properties
View file @
0d941848
# 模型唯一标识
modelCode
=
378
# 模型名称
# 模型名称
modelName
=
mobilenet_prune
modelName
=
mobilenet_prune
# 模型描述
# 模型描述
modelDescription
=
mobilenet_prune是一个实现对mobilenet、efficientnet剪枝的demo
modelDescription
=
mobilenet_prune是一个实现对mobilenet、efficientnet剪枝的demo
# 应用场景
# 应用场景
appScenario
=
剪枝、微调
appScenario
=
交通,金融,医疗,教育,家居
# 框架类型
# 框架类型
frameType
=
pytorch
frameType
=
pytorch
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment