训练源码:
- 源码仓库:https://github.com/pytorch/tutorials
- 迁移学习测试代码:tutorials/beginner_source/transfer_learning_tutorial.py
准备工作:
- 下载数数据集:https://download.pytorch.org/tutorial/hymenoptera_data.zip --> tutorials/beginner_source/data/hymenoptera_data
- 下载与训练模型:https://download.pytorch.org/models/resnet18-5c106cde.pth --> ~/.torch/models/resnet18-5c106cde.pth