实战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 提供通

>>阅读原文<<