利用pytorch對CIFAR-10數據集的分類

步驟以下: 1.使用torchvision加載並預處理CIFAR-10數據集、 2.定義網絡 3.定義損失函數和優化器 4.訓練網絡並更新網絡參數 5.測試網絡 運行環境: windows+python3.6.3+pycharm+pytorch0.3.0python import torchvision as tv import torchvision.transforms as transfor
相關文章
相關標籤/搜索