Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleSlim
提交
6410ed1a
P
PaddleSlim
项目概览
PaddlePaddle
/
PaddleSlim
大约 1 年 前同步成功
通知
51
Star
1434
Fork
344
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
53
列表
看板
标记
里程碑
合并请求
16
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleSlim
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
53
Issue
53
列表
看板
标记
里程碑
合并请求
16
合并请求
16
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
6410ed1a
编写于
12月 28, 2021
作者:
I
itminner
提交者:
GitHub
12月 28, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix quant aware with infer model readme (#963)
上级
e406740a
变更
1
显示空白变更内容
内联
并排
Showing
1 changed file
with
32 addition
and
18 deletion
+32
-18
demo/quant/quant_aware_with_infermodel/README.md
demo/quant/quant_aware_with_infermodel/README.md
+32
-18
未找到文件。
demo/quant/quant_aware_with_infermodel/README.md
浏览文件 @
6410ed1a
...
...
@@ -49,19 +49,16 @@ top1_acc/top5_acc= [0.71918 0.90568]
蒸馏量化训练示例脚本为
[
quant_aware_with_infermodel.py
](
./quant_aware_with_infermodel.py
)
,使用接口
``paddleslim.quant.quant_aware_with_infermodel``
对模型进行量化训练。运行命令为:
```
python quant_aware_with_infermodel.py \
--use_gpu=False \
--batch_size=2 \
--num_epoch=30 \
--save_iter_step=
10
0 \
--save_iter_step=
2
0 \
--learning_rate=0.0001 \
--weight_decay=0.00004 \
--use_pact=
Tru
e \
--use_pact=
Fals
e \
--checkpoint_path="./inference_model/MobileNet_quantaware_ckpt/" \
--model_path="./infermodel_mobilenetv2/" \
--model_filename="inference.pdmodel" \
--params_filename="inference.pdiparams" \
--teacher_model_path="./infermodel_mobilenetv2/" \
--teacher_model_filename="inference.pdmodel" \
--teacher_params_filename="inference.pdiparams" \
--model_path_prefix="./inference_model/infermodel_mobilenetv2/inference" \
--teacher_model_path_prefix="./inference_model/infermodel_mobilenetv2/inference" \
--distill_node_name_list "teacher_conv2d_54.tmp_0" "conv2d_54.tmp_0" "teacher_conv2d_55.tmp_0" "conv2d_55.tmp_0" \
"teacher_conv2d_57.tmp_0" "conv2d_57.tmp_0" "teacher_elementwise_add_0" "elementwise_add_0" \
"teacher_conv2d_61.tmp_0" "conv2d_61.tmp_0" "teacher_elementwise_add_1" "elementwise_add_1" \
...
...
@@ -80,12 +77,8 @@ python quant_aware_with_infermodel.py \
-
``weight_decay``
: 推荐使用float模型训练weight decay设置。
-
``use_pact``
: 是否使用pact量化算法, 推荐使用。
-
``checkpoint_path``
: 量化训练模型checkpoint保存路径。
-
``model_path``
: 需要量化的预测模型所在路径。
-
``model_filename``
: 如果需要量化的模型的参数文件保存在一个文件中,则设置为该模型的模型文件名称,如果参数文件保存在多个文件中,则不需要设置。
-
``params_filename``
: 如果需要量化的模型的参数文件保存在一个文件中,则设置为该模型的参数文件名称,如果参数文件保存在多个文件中,则不需要设置。
-
``teacher_model_path``
: teacher模型所在路径, 可以和量化模型是同一个,即自蒸馏。
-
``teacher_model_filename``
: teacher模型model文件名字。
-
``teacher_params_filename``
: teacher模型参数文件名字。
-
``model_path_prefix``
: 需要量化模型的目录 + 模型名称(不包含后缀)
-
``teacher_model_path_prefix``
: teacher模型的目录 + 模型名称(不包含后缀), 可以和量化模型是同一个,即自蒸馏。
-
``distill_node_name_list``
: 蒸馏节点名字列表,每两个节点组成一对,分别属于teacher模型和量化模型。
运行以上命令后,可在
``${checkpoint_path}``
下看到量化后模型的checkpoint。
...
...
@@ -96,12 +89,33 @@ python quant_aware_with_infermodel.py \
```
python export_quantmodel.py \
--use_gpu=True \
--checkpoint_path="./MobileNetV2_checkpoints/epoch_0_iter_2000" \
--infermodel_save_path="./quant_infermodel_mobilenetv2/" \
--use_gpu=False \
--batch_size=2 \
--num_epoch=30 \
--save_iter_step=20 \
--learning_rate=0.0001 \
--weight_decay=0.00004 \
--use_pact=False \
--checkpoint_path="./inference_model/MobileNet_quantaware_ckpt/" \
--model_path_prefix="./inference_model/infermodel_mobilenetv2/inference" \
--teacher_model_path_prefix="./inference_model/infermodel_mobilenetv2/inference" \
--distill_node_name_list "teacher_conv2d_54.tmp_0" "conv2d_54.tmp_0" "teacher_conv2d_55.tmp_0" "conv2d_55.tmp_0" \
"teacher_conv2d_57.tmp_0" "conv2d_57.tmp_0" "teacher_elementwise_add_0" "elementwise_add_0" \
"teacher_conv2d_61.tmp_0" "conv2d_61.tmp_0" "teacher_elementwise_add_1" "elementwise_add_1" \
"teacher_elementwise_add_2" "elementwise_add_2" "teacher_conv2d_67.tmp_0" "conv2d_67.tmp_0" \
"teacher_elementwise_add_3" "elementwise_add_3" "teacher_elementwise_add_4" "elementwise_add_4" \
"teacher_elementwise_add_5" "elementwise_add_5" "teacher_conv2d_75.tmp_0" "conv2d_75.tmp_0" \
"teacher_elementwise_add_6" "elementwise_add_6" "teacher_elementwise_add_7" "elementwise_add_7" \
"teacher_conv2d_81.tmp_0" "conv2d_81.tmp_0" "teacher_elementwise_add_8" "elementwise_add_8" \
"teacher_elementwise_add_9" "elementwise_add_9" "teacher_conv2d_87.tmp_0" "conv2d_87.tmp_0" \
"teacher_linear_1.tmp_0" "linear_1.tmp_0" \
--checkpoint_filename="epoch_0_iter_20" \
--export_inference_model_path_prefix="./inference_model/MobileNet_quantaware_ckpt/epoch_0_iter_20_infer"
```
-
``checkpoint_filename``
: checkpoint文件名。
-
``export_inference_model_path_prefix``
: 量化模型导出的目录 + 模型名称(不包含后缀)。
###5. 测试精度
###
5. 测试精度
使用
[
eval.py
](
../quant_post/eval.py
)
脚本对量化后的模型进行精度测试:
```
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录