训练源码:
- 源码仓库: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
原文地址:https://www.cnblogs.com/xbit/p/11324739.html
时间: 2025-01-17 10:32:53