从运行训练中保存模型 - Pytorch

时间:2021-04-18 08:07:52

标签: python pytorch

如果代码中没有提到,有没有办法从正在运行的脚本中保存模型?我在 GPU 上运行了 5 天以来的模型,并在达到指标的新高分时保存模型。不幸的是,数据集包含许多错误标记的样本。因此,尽管从定性的角度来看,该模型仍在取得进展,但目前得分正在下降。因此,模型不再保存。有没有办法保存当前模型,或者我必须再次运行它?

0 个答案:

没有答案