Colab(一0。二):PyTorch - Training a Classifier
2021/03/24
-----
說明:
由於 PyTorch 的 LeNet 範例 [1] 資料庫下載有問題,所以先練習 CIFAR10 [2]。完整的程式本次有跑完,參考附圖。[3] 在執行上有一些問題,可以參考 [2],本文下方有代碼呈現。
若要拷貝程式,可以使用 Tutorial 程式區右上角的拷貝標籤 [3]。
-----
# CIFAR10 影像分類。
-----
注意事項:
程式若出現 bug,可以修改代碼如下。
-----
transform = transforms.Compose(
[transforms.ToTensor()])
#transform = transforms.Compose(
# [transforms.ToTensor(),
# transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])
-----
References
[1] pytorch_quick_start.ipynb - Colaboratory
https://colab.research.google.com/drive/1gCgtlnMPVWY0l1ra8ssS0EzJiWaVR4Pk
[2] [DAY 05] 從頭訓練大Model?想多了 : Torchvision 簡介 - iT 邦幫忙::一起幫忙解決難題,拯救 IT 人的一天
https://ithelp.ithome.com.tw/articles/10218698
[3] Training a Classifier — PyTorch Tutorials 1.8.0 documentation
https://pytorch.org/tutorials/beginner/blitz/cifar10_tutorial.html
[4] cifar10_tutorial.ipynb - Colaboratory
-----
No comments:
Post a Comment