pytorch 学习笔记:nn.Sequential构造神经网络
生活随笔
收集整理的這篇文章主要介紹了
pytorch 学习笔记:nn.Sequential构造神经网络
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
1 兩種等價的構(gòu)造方式
1.1 構(gòu)建Net類:
1.2 使用torch.nn.Sequential
nn.Sequential里面的模塊按照順序進行排列的,所以必須確保前一個模塊的輸出大小和下一個模塊的輸入大小是一致的。
這兩種表達方式唯一的不同就是,在1.2中的方法中,激活函數(shù)也獨立算作一層。
2?OrderedDict
nn.Sequential中可以使用OrderedDict來指定每個module的名字,而不是采用默認的命名方式(按序號 0,1,2,3...)。
總結(jié)
以上是生活随笔為你收集整理的pytorch 学习笔记:nn.Sequential构造神经网络的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: pytorch实现简易分类模型
- 下一篇: 文巾解题 525. 连续数组