pytorch 测试 迁移学习

 

训练源码：

1. 源码仓库：https://github.com/pytorch/tutorials
2. 迁移学习测试代码：tutorials/beginner_source/transfer_learning_tutorial.py
   

 

准备工作：

1. 下载数数据集：https://download.pytorch.org/tutorial/hymenoptera_data.zip          -->  tutorials/beginner_source/data/hymenoptera_data
2. 下载与训练模型：https://download.pytorch.org/models/resnet18-5c106cde.pth    -->  ~/.torch/models/resnet18-5c106cde.pth

 

来源：https://www.cnblogs.com/xbit/p/11324739.html