pytorch微调与迁移学习

先迁移学习

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

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

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

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

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

在这里插入图片描述

  • 开始训练
    在这里插入图片描述
    预测精度提升了
    在这里插入图片描述
  • 保存最优的模型参数(即分数最高的) 来进行预测
    保存最优参数的介绍下面会介绍

猜你喜欢

转载自blog.csdn.net/weixin_41504611/article/details/128554474