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

猜你喜欢

转载自www.cnblogs.com/xbit/p/11324739.html