提交 57d252d5 编写于 作者: S ShusenTang

add remark in 3.3.4 #10

上级 3e8de6b4
......@@ -138,6 +138,8 @@ init.normal_(net[0].weight, mean=0, std=0.01)
init.constant_(net[0].bias, val=0) # 也可以直接修改bias的data: net[0].bias.data.fill_(0)
```
> 注:如果这里的`net`是用3.3.3节一开始的代码自定义的,那么上面代码会报错,`net[0].weight`应改为`net.linear.weight`,`bias`亦然。因为`net[0]`这样根据下标访问子模块的写法只有当`net`是个`ModuleList`或者`Sequential`实例时才可以,详见4.1节。
## 3.3.5 定义损失函数
PyTorch在`nn`模块中提供了各种损失函数,这些损失函数可看作是一种特殊的层,PyTorch也将这些损失函数实现为`nn.Module`的子类。我们现在使用它提供的均方误差损失作为模型的损失函数。
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册