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

雨点打透心脏的1/2处 2024-02-17 11:35 185阅读 0赞

最近由于网络的原因,在用 cascade rcnn 训练一个大型数据集的时候,在epoch= 20 的时候中断训练了。如何继续上次的训练过程?

打开 train.py,如图:

在这里插入图片描述

也就是说,训练时,最后加一个–resume from 参数,然后后面跟上次训练生成的最后一个权重文件(.pth)就可以了。

因此,在命令行输入以下语句进行训练即可以完全相同的模型配置继续上次的训练啦:

  1. python /home/wangzhenkuan/mmdetection/work_dirs/cascade-rcnn_r50_fpn_1x_coco.py --work-dir work_dirs --resume-from /home/wangzhenkuan/mmdetection/work_dirs/epoch_12.pth

发表评论

表情:
评论列表 (有 0 条评论,185人围观)

还没有评论,来说两句吧...

相关阅读

    相关 如何训练模型

    训练模型通常需要以下几个步骤: 1. 准备训练数据:这通常包括收集大量的输入数据和对应的正确输出,并将它们分成训练集和测试集。 2. 选择模型类型:根据问题的特点,选择合