Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
Models
提交
475b6b9f
M
Models
项目概览
MegEngine 天元
/
Models
通知
6
Star
3
Fork
3
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
Models
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
475b6b9f
编写于
6月 05, 2020
作者:
Y
Yizhuang Zhou
提交者:
GitHub
6月 05, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
feat(quantization): add pretrained weights and update accuracy of quantized models (#30)
上级
4de9258c
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
17 addition
and
4 deletion
+17
-4
official/quantization/README.md
official/quantization/README.md
+15
-2
official/quantization/config.py
official/quantization/config.py
+2
-2
未找到文件。
official/quantization/README.md
浏览文件 @
475b6b9f
...
@@ -6,13 +6,26 @@
...
@@ -6,13 +6,26 @@
| Model | top1 acc (float32) | FPS
* (float32) | top1 acc (int8) | FPS*
(int8) |
| Model | top1 acc (float32) | FPS
* (float32) | top1 acc (int8) | FPS*
(int8) |
| --- | --- | --- | --- | --- |
| --- | --- | --- | --- | --- |
| ResNet18 | 69.824 | 10.5 | 69.754 | 16.3 |
| ResNet18 | 69.824 | 10.5 | 69.754 | 16.3 |
| ShufflenetV1 (1.5x) | 71.954 | 17.3 | | 25.3 |
| ShufflenetV1 (1.5x) | 71.954 | 17.3 |
70.656
| 25.3 |
| MobilenetV2 | 72.820 | 13.1 | | 17.4 |
| MobilenetV2 | 72.820 | 13.1 |
71.378
| 17.4 |
**
: FPS is measured on Intel(R) Xeon(R) Gold 6130 CPU @ 2.10GHz, single 224x224 image
*
**
: FPS is measured on Intel(R) Xeon(R) Gold 6130 CPU @ 2.10GHz, single 224x224 image
*
*We finetune mobile models with QAT for 30 epochs, training longer may yield better accuracy*
量化模型使用时,统一读取0-255的uint8图片,减去128的均值,转化为int8,输入网络。
量化模型使用时,统一读取0-255的uint8图片,减去128的均值,转化为int8,输入网络。
#### (Optional) Download Pretrained Models
```
wget https://data.megengine.org.cn/models/weights/mobilenet_v2_normal_72820.pkl
wget https://data.megengine.org.cn/models/weights/mobilenet_v2_qat_71378.pkl
wget https://data.megengine.org.cn/models/weights/resnet18_normal_69824.pkl
wget https://data.megengine.org.cn/models/weights/resnet18_qat_69754.pkl
wget https://data.megengine.org.cn/models/weights/shufflenet_v1_x1_5_g3_normal_71954.pkl
wget https://data.megengine.org.cn/models/weights/shufflenet_v1_x1_5_g3_qat_70656.pkl
```
## Quantization Aware Training (QAT)
## Quantization Aware Training (QAT)
```
python
```
python
...
...
official/quantization/config.py
浏览文件 @
475b6b9f
...
@@ -48,8 +48,8 @@ def get_config(arch: str):
...
@@ -48,8 +48,8 @@ def get_config(arch: str):
class
ShufflenetFinetuneConfig
(
ShufflenetConfig
):
class
ShufflenetFinetuneConfig
(
ShufflenetConfig
):
BATCH_SIZE
=
128
//
2
BATCH_SIZE
=
128
//
2
LEARNING_RATE
=
0.0
3125
LEARNING_RATE
=
0.0
03125
//
2
EPOCHS
=
12
0
EPOCHS
=
3
0
class
ResnetFinetuneConfig
(
ResnetConfig
):
class
ResnetFinetuneConfig
(
ResnetConfig
):
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录