[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!
文章来源:https://www.toymoban.com/news/detail-484966.html
这样就可以避免重新训练浪费时间啦! 如果对你有用的话,就点个赞吧! 感恩的心!文章来源地址https://www.toymoban.com/news/detail-484966.html
到了这里,关于【 [mmdetection] 如何在训练中断后,接着上次训练?】的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!