pytorch 测试 迁移学习
Posted xbit
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了pytorch 测试 迁移学习相关的知识，希望对你有一定的参考价值。
训练源码：
- 源码仓库：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
以上是关于pytorch 测试 迁移学习的主要内容，如果未能解决你的问题，请参考以下文章