Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
FluidDoc
提交
ef940f19
F
FluidDoc
项目概览
PaddlePaddle
/
FluidDoc
通知
7
Star
2
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
23
列表
看板
标记
里程碑
合并请求
111
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
F
FluidDoc
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
23
Issue
23
列表
看板
标记
里程碑
合并请求
111
合并请求
111
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
ef940f19
编写于
9月 03, 2020
作者:
C
Chen Weihang
提交者:
GitHub
9月 03, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
refine set_state_dict of Layer (#2535)
上级
4054756f
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
13 addition
and
32 deletion
+13
-32
doc/paddle/api/paddle/fluid/dygraph/layers/Layer_cn.rst
doc/paddle/api/paddle/fluid/dygraph/layers/Layer_cn.rst
+13
-32
未找到文件。
doc/paddle/api/paddle/fluid/dygraph/layers/Layer_cn.rst
浏览文件 @
ef940f19
...
...
@@ -391,13 +391,14 @@ buffer是一个非参数类型的变量,不会被优化器更新,但在评
state_dict
=
emb
.
state_dict
()
fluid
.
save_dygraph
(
state_dict
,
"paddle_dy"
)
..
py
:
method
::
set_
dict
(
stat_dict
,
include_sublayers
=
True
)
..
py
:
method
::
set_
state_dict
(
state_dict
,
include_sublayers
=
True
,
use_structured_name
=
True
)
根据传入的
``
stat
_dict
``
设置参数和可持久性
buffers
。
所有参数和
buffers
将由
``
stat
_dict
``
中的
``
Tensor
``
设置。
根据传入的
``
stat
e_dict
``
设置参数和可持久性
buffers
。
所有参数和
buffers
将由
``
state
_dict
``
中的
``
Tensor
``
设置。
参数:
-
**
state_dict
**
(
dict
)
-
包含所有参数和可持久性
buffers
的
dict
。
-
**
include_sublayers
**
(
bool
,
可选
)
-
如果设置为
True
,则还包括子层的参数和
buffers
。
默认值:
True
。
-
**
use_structured_name
**
(
bool
,
可选
)
-
如果设置为
True
,将使用
Layer
的结构性变量名作为
dict
的
key
,否则将使用
Parameter
或者
Buffer
的变量名作为
key
。默认值:
True
。
返回:
None
...
...
@@ -405,36 +406,16 @@ buffer是一个非参数类型的变量,不会被优化器更新,但在评
..
code
-
block
::
python
import
paddle
.
fluid
as
fluid
with
fluid
.
dygraph
.
guard
():
emb
=
fluid
.
dygraph
.
Embedding
([
10
,
10
])
state_dict
=
emb
.
state_dict
()
fluid
.
save_dygraph
(
state_dict
,
"paddle_dy"
)
para_state_dict
,
_
=
fluid
.
load_dygraph
(
"paddle_dy"
)
emb
.
set_dict
(
para_state_dict
)
..
py
:
method
::
load_dict
(
stat_dict
,
include_sublayers
=
True
)
..
warning
::
该函数将被弃用。请使用
set_dict
函数。
根据传入的
``
stat_dict
``
设置参数和可持久性
buffers
。
所有参数和
buffers
将由
``
stat_dict
``
中的
``
Tensor
``
设置。
参数:
-
**
state_dict
**
(
dict
)
-
包含所有参数和可持久性
buffers
的
dict
。
-
**
include_sublayers
**
(
bool
,
可选
)
-
如果设置为
True
,则还包括子层的参数和
buffers
。
默认值:
True
。
返回:
None
import
paddle
paddle
.
disable_static
()
emb
=
paddle
.
nn
.
Embedding
([
10
,
10
])
**
代码示例
**
..
code
-
block
::
python
state_dict
=
emb
.
state_dict
()
paddle
.
save
(
state_dict
,
"paddle_dy"
)
para_state_dict
,
_
=
paddle
.
load
(
"paddle_dy"
)
import
paddle
.
fluid
as
fluid
with
fluid
.
dygraph
.
guard
():
emb
=
fluid
.
dygraph
.
Embedding
([
10
,
10
])
state_dict
=
emb
.
state_dict
()
fluid
.
save_dygraph
(
state_dict
,
"paddle_dy"
)
para_state_dict
,
_
=
fluid
.
load_dygraph
(
"paddle_dy"
)
emb
.
load_dict
(
para_state_dict
)
emb
.
set_state_dict
(
para_state_dict
)
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录