【python模型保存之后，再恢复训练如何做】

权侑莉老婆

已于 2022-06-03 15:22:56 修改

阅读量1.5k

点赞数

文章标签： python pytorch 深度学习

于 2022-06-02 15:04:41 首次发布

本文链接：https://blog.youkuaiyun.com/m0_54487794/article/details/125100739

版权

在PyTorch中，中断训练并保存模型后，恢复训练时准确率未提升。问题在于重新加载模型参数后，直接使用之前的优化器是错误的。正确做法是重新初始化优化器，以使训练能够继续进行，这样准确率才会继续增加。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

实际上，想要中断训练，然后保存模型，之后再加载继续训练，并没有那么简单。我们平常保存模型参数一般都是这样遍历之后，再保存到字典里

    for name, data in global_model.state_dict().items():
        diff[name] = (data)
        c = diff[name]