Pytorch系列之——模型創建與nn.Module

模型創建與nn.Module 網絡模型創建步驟 nn.Module屬性 在學習模型創建步驟之前，回顧一下前面提到的機器學習模型訓練的五大步驟，即：數據、模型、損失函數、優化器和迭代訓練。這次我們主要談論模型模塊這部分。 對於模型模塊，首先分爲兩大部分分別是模型的創建和模型權值的初始化。其中在模型創建模塊，有分爲構建網絡層和拼接網絡層這兩個子模塊，我們通過構建類似卷積層、池化層和激活函數層等子模塊，

>>阅读原文<<