本文重點

ResNet模型已經搭建完成了，本文我們使用ResNet來跑一下CAFIR10的數據集，看一下分類效果如何？

代碼

本文總結

在之前的課程中我們對殘差塊以及ResNet模型進行了詳細的介紹，并且我們對模型訓練這些基礎的數據集進行了詳細的使用，我們只需要加載好模型，然后將數據以特定的格式輸入到模型中，此時前向傳播完成，然后將前向傳播的輸出和真實的標簽計算交叉熵損失，然后進行反向傳播。至此一段訓練完成，經過多次訓練之后，最終完成模型的訓練，最終我們將訓練曲線畫出來。