[mmdetection] 如何在训练中断后,接着上次训练?

最近由于不知名原因,在用 faster rcnn 训练一个大型数据集的时候,在epoch= 20 的时候中断训练了.采用以下方式继续上次训练.
打开 train.py,如图:

也就是说,训练时,最后加一个–resume from 参数,然后后面跟上次训练生成的最后一个权重文件(.pth)就可以了.
因此,在命令行输入以下语句进行训练即可以完全相同的模型配置继续上次的训练啦::

 python tools/train.py configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py --work-dir work_dirs/xxx --resume-from work_dirs/yyy/latest.pth

运行之后,就会发现直接从 epoch=21 开始输出训练结果啦! happy!

这样就可以避免重新训练浪费时间啦! 如果对你有用的话,就点个赞吧! 感恩的心!