pytorch微调与迁移学习

文章目录

先迁移学习

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

  • 在实际应用中，一般解冻靠近输出的卷积
    下面展示解冻全部卷积的部分：
    在这里插入图片描述
    在这里插入图片描述
    然后开始训练，训练完以后查看精度

  • 最终结果在94.多%
    在这里插入图片描述

下面进行微调来提升模型精度

  • 全连接层全部训练好以后才能微调（重点，一定要记住）
    在这里插入图片描述

在这里插入图片描述

  • 开始训练
    在这里插入图片描述
    预测精度提升了
    在这里插入图片描述
  • 保存最优的模型参数（即分数最高的） 来进行预测
    保存最优参数的介绍下面会介绍
版权声明：本文为weixin_41504611原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。