Pytorch學習 model類

model是如何作用的 model的兩個要素 1.構建子模塊： model的初始化函數 init(),要使用supper調用nn.model類，在初始化函數中我們構建子模塊。 2. 拼接子模塊： model 的 forward函數中我們用子模塊構建計算圖，返回計算結果 在model類的引用處設置斷點 步入leNet中 進入定義leNet的model文件中： 模型構建第一個要素，構建子模塊，就是在m

>>阅读原文<<