训练中断后,怎样加载已经训练的权重
收藏
在飞桨运行deformer-detr,运行到49epoch后就中断了,能够找到最后保存的训练权重文件“output/xilanhua_deformable_detr/model_final.pdparams”,请问应该在哪里修改(命令行里还是需要在yml文件里),能够加载这个权重继续训练呢?盼望指导,谢谢大家!
0
收藏
请登录后评论
直接把这个当做预训练模型加载吧,仔细查看一下文档吧。
好的谢谢!
加油小伙子
学到了
--resume_model 预训练的位置
--resume也可以