专栏算法工具链模型训练中断后继续训练，使用哪个pth

模型训练中断后继续训练，使用哪个pth

已解决
Robin。2025-03-17
28
4

如上图，生成了000*.pth、last-2ac20817.pth、last.pth，继续训练的话，该选用哪个文件保存的权重呢？

算法工具链
征程5
评论1
0/1000
  • Huanghui
    Lv.5

    你好，应该是最后一个 ，你加载下试试看

    2025-03-17
    0
    3
    • Robin。回复Huanghui:
      好的，谢谢。
      请问这个last-2ac20817和last之间的区别在哪呢
      2025-03-17
      0
    • Huanghui回复Robin。:

      可能是介于开始和中断之间轮次的pt权重。后缀都为pth，说明都是权重，中间的数字应该只是某个checkpoint的标记吧。我的理解是这样的

      2025-03-17
      0
    • Robin。回复Huanghui:

      好的，感谢～

      2025-03-17
      0