Wednesday, March 24, 2021

Colab(一0。二):PyTorch - Training a Classifier

 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

https://colab.research.google.com/github/pytorch/tutorials/blob/gh-pages/_downloads/17a7c7cb80916fcdf921097825a0f562/cifar10_tutorial.ipynb

-----

No comments: