Skip to content

Commit 5e9fbaf

Browse files
author
Nyako Shigure
committed
Add Chinese docs for Optimizer.state_dict/load_state_dict/set_state_dict and fix links
- Add state_dict(), set_state_dict(), load_state_dict() methods to Optimizer_cn.rst - Update state_dict mapping link to Chinese docs - Update load_state_dict mapping link to Chinese docs (from #7782)
1 parent af9ccd2 commit 5e9fbaf

3 files changed

Lines changed: 49 additions & 2 deletions

File tree

docs/api/paddle/optimizer/Optimizer_cn.rst

Lines changed: 47 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -124,6 +124,53 @@ set_lr_scheduler(scheduler)
124124

125125
COPY-FROM: paddle.optimizer.Optimizer.set_lr_scheduler
126126

127+
state_dict()
128+
'''''''''
129+
130+
获取优化器的状态字典信息。包含优化器使用的所有 Tensor。对于 Adam 优化器,包含 beta1、beta2、momentum 等。如果使用了 LRScheduler,状态字典中还会包含 global_step。如果优化器从未被调用过(minimize 函数),则状态字典为空。
131+
132+
**返回**
133+
134+
dict[str, Tensor],包含优化器使用的所有 Tensor 的字典。
135+
136+
**代码示例**
137+
138+
COPY-FROM: paddle.optimizer.Optimizer.state_dict
139+
140+
set_state_dict(state_dict)
141+
'''''''''
142+
143+
加载优化器的状态字典。对于 Adam 优化器,包含 beta1、beta2、momentum 等。如果使用了 LRScheduler,global_step 也将被更新。
144+
145+
**参数**
146+
147+
- **state_dict** (dict) - 包含优化器所需的所有 Tensor 的字典。
148+
149+
**返回**
150+
151+
无。
152+
153+
**代码示例**
154+
155+
COPY-FROM: paddle.optimizer.Optimizer.set_state_dict
156+
157+
load_state_dict(state_dict)
158+
'''''''''
159+
160+
加载优化器的状态字典。对于 Adam 优化器,包含 beta1、beta2、momentum 等。如果使用了 LRScheduler,global_step 也将被更新。
161+
162+
**参数**
163+
164+
- **state_dict** (dict) - 包含优化器所需的所有 Tensor 的字典。
165+
166+
**返回**
167+
168+
无。
169+
170+
**代码示例**
171+
172+
COPY-FROM: paddle.optimizer.Optimizer.load_state_dict
173+
127174
get_lr()
128175
'''''''''
129176

docs/guides/model_convert/convert_from_pytorch/api_difference/invok_only_diff/torch.optim.Optimizer.load_state_dict.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
torch.optim.Optimizer.load_state_dict(state_dict)
77
```
88

9-
### [paddle.optimizer.Optimizer.load_state_dict](https://www.paddlepaddle.org.cn/documentation/docs/en/develop/api/paddle/optimizer/Optimizer_en.html#set_state_dict)
9+
### [paddle.optimizer.Optimizer.load_state_dict](https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/optimizer/Optimizer_cn.html#load-state-dict-state-dict)
1010

1111
```python
1212
paddle.optimizer.Optimizer.load_state_dict(state_dict)

docs/guides/model_convert/convert_from_pytorch/api_difference/invok_only_diff/torch.optim.Optimizer.state_dict.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
torch.optim.Optimizer.state_dict()
77
```
88

9-
### [paddle.optimizer.Optimizer.state_dict](https://www.paddlepaddle.org.cn/documentation/docs/en/api/paddle/optimizer/Optimizer_en.html#state_dict)
9+
### [paddle.optimizer.Optimizer.state_dict](https://www.paddlepaddle.org.cn/documentation/docs/zh/api/paddle/optimizer/Optimizer_cn.html#state-dict)
1010

1111
```python
1212
paddle.optimizer.Optimizer.state_dict()

0 commit comments

Comments
 (0)