實戰PyTorch（二）：寶可夢自定義數據集之遷移學習

使用數據集：寶可夢【皮卡丘：234，超夢：239；傑尼龜：223；小火龍：238；妙蛙種子：234】 思想：數據較少，能夠利用遷移學習來得到好的效果。python 4 steps數組 Load data Build model Train and Test Transfer Learning ▪ Load data Inherit from torch.utils.data.Dataset 提供通

>>阅读原文<<