本文中代码来自pyTorch官网中关于迁移学习的代码示例。 如需更详细学习迁移学习的内容,请点击上面代码示例超链接。 1. 模块引入 2. 数据准备 3. 展示部分数据 4. 训练模型 5. 测试训练结果 6. 加载预训练模型训练——版本1 7. 加载预训练模型训练——版本2