【深度之眼】Pytorch框架班第五期-模型創建機制

調試 1、以LeNet爲例,我們在第72行設置斷點,並進行Debug。 2、點擊C按鈕進入到模型中,我們可以發現__init__()函數構建了LeNet模型需要的子模塊。 3、我們一直點擊B按鈕,當最後一個模塊構建完成後,程序跳出,LeNet模型的初始化就完成了。模型構建完成。 4、我們將代碼運行到模型訓練部分的97行,此處我們對模型輸入了inputs,輸出outputs。 5、點擊C按鈕進入該行
相關文章
相關標籤/搜索