Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

refine state dict hook cn doc #4923

Merged
merged 10 commits into from
Jun 15, 2022
Merged

refine state dict hook cn doc #4923

Changes from all commits
Commits
Show all changes
10 commits
Select commit Hold shift + click to select a range
2899b2a
refine paddle.amp.decorate input parameter of optimizer
zhangbo9674 Dec 9, 2021
7a728f7
Revert "refine paddle.amp.decorate input parameter of optimizer"
zhangbo9674 Dec 9, 2021
62a4020
Merge branch 'develop' of https://github.com/PaddlePaddle/docs into d…
zhangbo9674 Dec 24, 2021
95d9ed0
Merge branch 'develop' of https://github.com/PaddlePaddle/docs into d…
zhangbo9674 Mar 4, 2022
9d76c0b
Merge branch 'develop' of https://github.com/PaddlePaddle/docs into d…
zhangbo9674 May 30, 2022
a2009bb
Merge branch 'develop' of https://github.com/PaddlePaddle/docs into d…
zhangbo9674 Jun 6, 2022
b0af2ef
Merge branch 'develop' of https://github.com/PaddlePaddle/docs into d…
zhangbo9674 Jun 14, 2022
282c71e
refine cn doc for state_dict
zhangbo9674 Jun 14, 2022
d6c1c47
refine doc
zhangbo9674 Jun 14, 2022
13bf8f3
refine doc
zhangbo9674 Jun 14, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion docs/api/paddle/nn/Layer_cn.rst
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -658,7 +658,7 @@ Parameter， 传入的参数实例
print(name, param) # will print w_tmp,_linear.weight,_linear.bias


state_dict(destination=None, include_sublayers=True)
state_dict(destination=None, include_sublayers=True, use_hook=True)
'''''''''

获取当前层及其子层的所有参数和可持久性buffers。并将所有参数和buffers存放在dict结构中。
Expand All @@ -667,6 +667,7 @@ state_dict(destination=None, include_sublayers=True)

- **destination** (dict, 可选) - 如果提供 ``destination`` ，则所有参数和可持久性buffers都将存放在 ``destination`` 中。 默认值：None。
- **include_sublayers** (bool, 可选) - 如果设置为True，则包括子层的参数和buffers。默认值：True。
- **use_hook** (bool, 可选) - 如果设置为True，将_state_dict_hooks中注册的函数应用于destination。默认值：True。

**返回**
dict， 包含所有参数和可持久行buffers的dict
Expand Down