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

 

posted @ 2019-08-09 00:37  xbit  阅读(527)  评论(0编辑  收藏  举报