Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleSlim
提交
63140aed
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看板
未验证
提交
63140aed
编写于
5月 18, 2022
作者:
C
Chang Xu
提交者:
GitHub
5月 18, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add HyperParams demo for Prune and Distillation (#1112)
上级
9c214906
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
32 addition
and
2 deletion
+32
-2
demo/auto_compression/README.md
demo/auto_compression/README.md
+1
-1
demo/auto_compression/hyperparameter_tutorial.md
demo/auto_compression/hyperparameter_tutorial.md
+31
-1
未找到文件。
demo/auto_compression/README.md
浏览文件 @
63140aed
...
...
@@ -9,7 +9,7 @@ PaddleSlim推出全新自动压缩工具(ACT),旨在通过Source-Free的
## 环境准备
-
安装PaddlePaddle >= 2.
3
版本 (从
[
Paddle官网
](
https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/install/pip/linux-pip.html
)
下载安装)
-
安装PaddlePaddle >= 2.
2
版本 (从
[
Paddle官网
](
https://www.paddlepaddle.org.cn/install/quick?docurl=/documentation/docs/zh/install/pip/linux-pip.html
)
下载安装)
-
安装PaddleSlim >= 2.3 或者适当develop版本
## 快速上手
...
...
demo/auto_compression/hyperparameter_tutorial.md
浏览文件 @
63140aed
...
...
@@ -20,20 +20,50 @@ Quantization:
#### 配置定制蒸馏策略
蒸馏参数主要设置蒸馏节点(
`distill_node_pair`
)和教师预测模型路径
。蒸馏节点需包含教师网络节点和对应的学生网络节点,其中教师网络节点名称将在程序中自动添加 “teacher_” 前缀
,如下所示:
蒸馏参数主要设置蒸馏节点(
`distill_node_pair`
)和教师预测模型路径,如下所示:
```
yaml
Distillation
:
# distill_lambda: distill loss所占权重;可输入多个数值,支持不同节点之间使用不同的lambda值
distill_lambda
:
1.0
# distill_loss: 蒸馏loss算法;可输入多个loss,支持不同节点之间使用不同的loss算法
distill_loss
:
l2_loss
# distill_node_pair: 蒸馏节点,即某层输出的变量名称,需包含教师网络节点和对应的学生网络节点,
# 其中教师网络节点名称将在程序中自动添加 “teacher_” 前缀;
# 可输入多个node_pair,支持多节点蒸馏
distill_node_pair
:
-
teacher_relu_30.tmp_0
-
relu_30.tmp_0
# merge_feed: 若teacher和student的输入相同则为true,若teacher和student的输入不同则为false
merge_feed
:
true
# teacher_model_dir: 保存预测模型文件和预测模型参数文件的文件夹名称
teacher_model_dir
:
./inference_model
# teacher_model_filename: 预测模型文件,格式为 *.pdmodel 或 __model__
teacher_model_filename
:
model.pdmodel
# teacher_params_filename: 预测模型参数文件,格式为 *.pdiparams 或 __params__
teacher_params_filename
:
model.pdiparams
```
-
蒸馏loss目前支持的有:fsp_loss,l2_loss,soft_label_loss,也可自定义loss。具体定义和使用可参考
[
知识蒸馏API文档
](
https://paddleslim.readthedocs.io/zh_CN/latest/api_cn/static/dist/single_distiller_api.html
)
。
#### 配置定制结构化稀疏策略
结构化稀疏参数设置如下所示:
```
yaml
Prune
:
# prune_algo: 裁剪算法
prune_algo
:
prune
# pruned_ratio: 裁剪比例
pruned_ratio
:
0.25
# prune_params_name: 需要裁剪的参数名字
prune_params_name
:
-
conv1_weights
# criterion: 评估一个卷积层内通道重要性所参考的指标
criterion
:
l1_norm
```
-
prune_algo目前支持的有:prune、asp和transformer_pruner。
-
criterion目前支持的有:l1_norm , bn_scale , geometry_median。具体定义和使用可参考
[
结构化稀疏API文档
](
https://paddleslim.readthedocs.io/zh_CN/latest/api_cn/static/prune/prune_api.html
)
。
#### 配置定制非结构化稀疏策略
非结构化稀疏参数设置如下所示,其中参数含义详见
[
非结构化稀疏API文档
](
https://github.com/PaddlePaddle/PaddleSlim/blob/develop/docs/zh_cn/api_cn/dygraph/pruners/unstructured_pruner.rst
)
:
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录