pytorch 学习笔记:nn.Sequential构造神经网络
生活随笔
收集整理的这篇文章主要介绍了
pytorch 学习笔记:nn.Sequential构造神经网络
小编觉得挺不错的,现在分享给大家,帮大家做个参考.
1 两种等价的构造方式
1.1 构建Net类:
1.2 使用torch.nn.Sequential
nn.Sequential里面的模块按照顺序进行排列的,所以必须确保前一个模块的输出大小和下一个模块的输入大小是一致的。
这两种表达方式唯一的不同就是,在1.2中的方法中,激活函数也独立算作一层。
2 OrderedDict
nn.Sequential中可以使用OrderedDict来指定每个module的名字,而不是采用默认的命名方式(按序号 0,1,2,3...)。
总结
以上是生活随笔为你收集整理的pytorch 学习笔记:nn.Sequential构造神经网络的全部内容,希望文章能够帮你解决所遇到的问题。
- 上一篇: pytorch实现简易分类模型
- 下一篇: 文巾解题 525. 连续数组