網絡模型創建的步驟及nn.Module屬性

  在數據模塊中我們瞭解了pytorch是如何從硬盤中讀取數據,然後對數據進行預處理,最後轉化爲張量的形式輸入到我們的模型當中。   在模型裏,是對數據模塊出處的張量進行復雜的數學運算,最後得到可以分類、分割、目標檢測的輸出。     一、網絡模型創建的步驟 這一節學習模型模塊。     模型模塊又分兩個模塊,一個是模型創建,一個是權值初始化。 在模型創建中,又分爲兩個子模塊。構建網絡層和拼接網絡
相關文章
相關標籤/搜索