Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleSlim
提交
c953571d
P
PaddleSlim
项目概览
PaddlePaddle
/
PaddleSlim
大约 2 年 前同步成功
通知
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看板
提交
c953571d
编写于
12月 20, 2019
作者:
B
baiyfbupt
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix doc
上级
1fc102c7
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
8 addition
and
8 deletion
+8
-8
docs/docs/api/single_distiller_api.md
docs/docs/api/single_distiller_api.md
+8
-8
未找到文件。
docs/docs/api/single_distiller_api.md
浏览文件 @
c953571d
...
@@ -5,12 +5,12 @@ paddleslim.dist.merge(teacher_program, student_program, data_name_map, place, sc
...
@@ -5,12 +5,12 @@ paddleslim.dist.merge(teacher_program, student_program, data_name_map, place, sc
**参数:**
**参数:**
-
**teacher_program**
(Program)-定义了teacher模型的
***[paddle program](https://www.paddlepaddle.org.cn/documentation/docs/zh/api_cn/fluid_cn/Program_cn.html#program)**
*
-
**teacher_program**
(Program)-定义了teacher模型的
[
*paddle program*
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/api_cn/fluid_cn/Program_cn.html#program
)
-
**student_program**
(Program)-定义了student模型的
***[paddle program](https://www.paddlepaddle.org.cn/documentation/docs/zh/api_cn/fluid_cn/Program_cn.html#program)**
*
-
**student_program**
(Program)-定义了student模型的
[
*paddle program*
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/api_cn/fluid_cn/Program_cn.html#program
)
-
**data_name_map**
(dict)-teacher输入接口名与student输入接口名的映射,其中dict的
*key*
为teacher的输入名,
*value*
为student的输入名
-
**data_name_map**
(dict)-teacher输入接口名与student输入接口名的映射,其中dict的
*key*
为teacher的输入名,
*value*
为student的输入名
-
**place**
(fluid.CPUPlace()|fluid.CUDAPlace(N))-该参数表示程序运行在何种设备上,这里的N为GPU对应的ID
-
**place**
(fluid.CPUPlace()|fluid.CUDAPlace(N))-该参数表示程序运行在何种设备上,这里的N为GPU对应的ID
-
**scope**
(Scope)-该参数表示程序使用的变量作用域,如果不指定将使用默认的全局作用域。默认值:
***[fluid.global_scope()](https://www.paddlepaddle.org.cn/documentation/docs/zh/api_cn/fluid_cn/global_scope_cn.html#global-scope)**
*
-
**scope**
(Scope)-该参数表示程序使用的变量作用域,如果不指定将使用默认的全局作用域。默认值:
[
*fluid.global_scope()*
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/api_cn/fluid_cn/global_scope_cn.html#global-scope
)
-
**name_prefix**
(str)-merge操作将统一为teacher的
[
Variables
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/1.3/api_guides/low_level/program.html#variable
)
添加的名称前缀name_prefix。默认值:'teacher_'
-
**name_prefix**
(str)-merge操作将统一为teacher的
[
*Variables*
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/1.3/api_guides/low_level/program.html#variable
)
添加的名称前缀name_prefix。默认值:'teacher_'
**返回:**
由student_program和teacher_program merge得到的program
**返回:**
由student_program和teacher_program merge得到的program
...
@@ -53,7 +53,7 @@ paddleslim.dist.fsp_loss(teacher_var1_name, teacher_var2_name, student_var1_name
...
@@ -53,7 +53,7 @@ paddleslim.dist.fsp_loss(teacher_var1_name, teacher_var2_name, student_var1_name
-
**teacher_var2_name**
(str): teacher_var2的名称. 对应的variable是一个形为
`[batch_size, y_channel, height, width]`
的4-D特征图Tensor,数据类型为float32或float64。只有y_channel可以与teacher_var1的x_channel不同,其他维度必须与teacher_var1相同
-
**teacher_var2_name**
(str): teacher_var2的名称. 对应的variable是一个形为
`[batch_size, y_channel, height, width]`
的4-D特征图Tensor,数据类型为float32或float64。只有y_channel可以与teacher_var1的x_channel不同,其他维度必须与teacher_var1相同
-
**student_var1_name**
(str): student_var1的名称. 对应的variable需与teacher_var1尺寸保持一致,是一个形为
`[batch_size, x_channel, height, width]`
的4-D特征图Tensor,数据类型为float32或float64
-
**student_var1_name**
(str): student_var1的名称. 对应的variable需与teacher_var1尺寸保持一致,是一个形为
`[batch_size, x_channel, height, width]`
的4-D特征图Tensor,数据类型为float32或float64
-
**student_var2_name**
(str): student_var2的名称. 对应的variable需与teacher_var2尺寸保持一致,是一个形为
`[batch_size, y_channel, height, width]`
的4-D特征图Tensor,数据类型为float32或float64。只有y_channel可以与student_var1的x_channel不同,其他维度必须与student_var1相同
-
**student_var2_name**
(str): student_var2的名称. 对应的variable需与teacher_var2尺寸保持一致,是一个形为
`[batch_size, y_channel, height, width]`
的4-D特征图Tensor,数据类型为float32或float64。只有y_channel可以与student_var1的x_channel不同,其他维度必须与student_var1相同
-
**program**
(Program): 用于蒸馏训练的fluid program。默认值:
[
fluid.default_main_program()
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/1.3/api_cn/fluid_cn.html#default-main-program
)
-
**program**
(Program): 用于蒸馏训练的fluid program。默认值:
[
*fluid.default_main_program()*
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/1.3/api_cn/fluid_cn.html#default-main-program
)
**返回:**
由teacher_var1, teacher_var2, student_var1, student_var2组合得到的fsp_loss
**返回:**
由teacher_var1, teacher_var2, student_var1, student_var2组合得到的fsp_loss
...
@@ -93,7 +93,7 @@ paddleslim.dist.l2_loss(teacher_var_name, student_var_name, program=fluid.defaul
...
@@ -93,7 +93,7 @@ paddleslim.dist.l2_loss(teacher_var_name, student_var_name, program=fluid.defaul
-
**teacher_var_name**
(str): teacher_var的名称.
-
**teacher_var_name**
(str): teacher_var的名称.
-
**student_var_name**
(str): student_var的名称.
-
**student_var_name**
(str): student_var的名称.
-
**program**
(Program): 用于蒸馏训练的fluid program。默认值:
[
fluid.default_main_program()
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/1.3/api_cn/fluid_cn.html#default-main-program
)
-
**program**
(Program): 用于蒸馏训练的fluid program。默认值:
[
*fluid.default_main_program()*
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/1.3/api_cn/fluid_cn.html#default-main-program
)
**返回:**
由teacher_var, student_var组合得到的l2_loss
**返回:**
由teacher_var, student_var组合得到的l2_loss
...
@@ -133,7 +133,7 @@ paddleslim.dist.soft_label_loss(teacher_var_name, student_var_name, program=flui
...
@@ -133,7 +133,7 @@ paddleslim.dist.soft_label_loss(teacher_var_name, student_var_name, program=flui
-
**teacher_var_name**
(str): teacher_var的名称.
-
**teacher_var_name**
(str): teacher_var的名称.
-
**student_var_name**
(str): student_var的名称.
-
**student_var_name**
(str): student_var的名称.
-
**program**
(Program): 用于蒸馏训练的fluid program。默认值:
[
fluid.default_main_program()
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/1.3/api_cn/fluid_cn.html#default-main-program
)
-
**program**
(Program): 用于蒸馏训练的fluid program。默认值:
[
*fluid.default_main_program()*
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/1.3/api_cn/fluid_cn.html#default-main-program
)
-
**teacher_temperature**
(float): 对teacher_var进行soft操作的温度值,温度值越大得到的特征图越平滑
-
**teacher_temperature**
(float): 对teacher_var进行soft操作的温度值,温度值越大得到的特征图越平滑
-
**student_temperature**
(float): 对student_var进行soft操作的温度值,温度值越大得到的特征图越平滑
-
**student_temperature**
(float): 对student_var进行soft操作的温度值,温度值越大得到的特征图越平滑
...
@@ -174,7 +174,7 @@ paddleslim.dist.loss(loss_func, program=fluid.default_main_program(), **kwargs)
...
@@ -174,7 +174,7 @@ paddleslim.dist.loss(loss_func, program=fluid.default_main_program(), **kwargs)
**参数:**
**参数:**
-
**loss_func**
(python function): 自定义的损失函数,输入为teacher var和student var,输出为自定义的loss
-
**loss_func**
(python function): 自定义的损失函数,输入为teacher var和student var,输出为自定义的loss
-
**program**
(Program): 用于蒸馏训练的fluid program。默认值:
[
fluid.default_main_program()
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/1.3/api_cn/fluid_cn.html#default-main-program
)
-
**program**
(Program): 用于蒸馏训练的fluid program。默认值:
[
*fluid.default_main_program()*
](
https://www.paddlepaddle.org.cn/documentation/docs/zh/1.3/api_cn/fluid_cn.html#default-main-program
)
-
**\**kwargs**
: loss_func输入名与对应variable名称
-
**\**kwargs**
: loss_func输入名与对应variable名称
**返回**
:自定义的损失函数loss
**返回**
:自定义的损失函数loss
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录