未验证 提交 62914ade 编写于 作者: X xiaotinghe 提交者: GitHub

fix d2lbook build (#1012)

Co-authored-by: NUbuntu <ubuntu@ip-172-31-12-66.us-west-2.compute.internal>
上级 64c831fe
......@@ -275,7 +275,7 @@ def evaluate_loss(data_iter, net, devices):
l = loss(outputs, labels)
l_sum += l.sum()
n += labels.numel()
return l_sum / n
return (l_sum / n).to('cpu')
```
## 定义[**训练函数**]
......
......@@ -120,6 +120,10 @@ net(X)
[Discussions](https://discuss.d2l.ai/t/1834)
:end_tab:
:begin_tab:`pytorch`
[Discussions](https://discuss.d2l.ai/t/1834)
:end_tab:
:begin_tab:`tensorflow`
[Discussions](https://discuss.d2l.ai/t/1833)
:end_tab:
......@@ -144,3 +144,7 @@ print(segment_BPE(tokens, symbols))
:begin_tab:`mxnet`
[Discussions](https://discuss.d2l.ai/t/386)
:end_tab:
:begin_tab:`pytorch`
[Discussions](https://discuss.d2l.ai/t/386)
:end_tab:
\ No newline at end of file
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册