PyTorch 迁移学习实践(几分钟即可训练好自己的模型)

前言如果你认为深度学习非常的吃GPU，或者说非常的耗时间，训练一个模型要非常久，但是你如果了解了迁移学习方法介绍。需要导入的包数据准备本次实验的数据到这里选择预训练的模型这里我们选择了使用model.buffers查看网络基本结构我们现在需要做的就是将最后一层进行替换训练主体的实现在使用cpu训练的情况，也能快速得到较好的结果，这里训练了50次，其实很快的就已经得到了很好的结果了总结本节我们使用了