Skip to content

Commit 5e5e28a

Browse files
幸苦了! Merge pull request #688 from ice-world/master
补全了缺失的部分代码
2 parents 6aebc34 + bfd1eb2 commit 5e5e28a

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

docs/2.0/tutorials/beginner/basics/optimization_tutorial.md

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -212,6 +212,18 @@ def test_loop(dataloader, model, loss_fn):
212212

213213
我们初始化了损失函数和优化器,传递给 `train_loop``test_loop`。你可以随意地修改 epochs 的数量来跟踪模型表现的进步情况。
214214

215+
```py
216+
loss_fn = nn.CrossEntropyLoss()
217+
optimizer = torch.optim.SGD(model.parameters(), lr=learning_rate)
218+
219+
epochs = 10
220+
for t in range(epochs):
221+
print(f"Epoch {t+1}\n-------------------------------")
222+
train_loop(train_dataloader, model, loss_fn, optimizer)
223+
test_loop(test_dataloader, model, loss_fn)
224+
print("Done!")
225+
```
226+
215227
输出:
216228

217229
```py

0 commit comments

Comments
 (0)