Skip to content

GitLab

Commit 0e76ddfb authored by dcuai's avatar dcuai
Browse files

Update README.md

parent dc59df31
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
README.md
View file @ 0e76ddfb
...@@ -15,7 +15,7 @@ mobilenetv2: ...@@ -15,7 +15,7 @@ mobilenetv2:
## 算法原理 ## 算法原理
mobilenetv2: mobilenetv2:
具有线性瓶颈的反转残差(inverted residual )：该模块将低维压缩表示作为输入，首先将其扩展为高维度并使用轻量级沿深度卷积(depthwiseconvolution)进行滤波。随后通过线性卷积将特征投射回低维表示。中间层使用轻量级的沿深度卷积来对特征进行滤波作为非线性的来源。 具有线性瓶颈的反转残差(inverted residual )：该模块将低维压缩表示作为输入，首先将其扩展为高维度并使用轻量级沿深度卷积(depthwiseconvolution)进行滤波。随后通过线性卷积将特征投射回低维表示。中间层使用轻量级的沿深度卷积来对特征进行滤波作为非线性的来源。
![image](https://developer.hpccube.com/codes/modelzoo/mobilenet_prune/-/blob/main/mobilenetV2.png) ![image](https://developer.hpccube.com/codes/modelzoo/mobilenet_prune/-/raw/main/mobilenetV2.png)
## 数据集 ## 数据集
`CIFAR-10` `CIFAR-10`
...@@ -75,7 +75,7 @@ mobilenetv3 ...@@ -75,7 +75,7 @@ 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 ## result
![image](https://developer.hpccube.com/codes/modelzoo/mobilenet_prune/-/blob/main/result.png) ![image](https://developer.hpccube.com/codes/modelzoo/mobilenet_prune/-/raw/main/result.png)
### 精度 ### 精度
method l1 method l1
| model | Base Acc | Pruned Acc | | model | Base Acc | Pruned Acc |
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment