断电等问题时常发生,因此我们会对模型进行断点训练。
如mmdetection项目中的FasterRCNN模型训练时出现训练突然中断,我们将使用如下命令从中断的epochs继续训练。
首先我们来看看模型训练时会生成的的文件:
接下来我们将使用如下命令进行断点训练:
python tools/train.py configs/faster_rcnn/faster-rcnn_r101_fpn_2x_coco.py --work-dir models/mmdetection/work_dirs/faster-rcnn_r101_fpn_2x_coco/20240721_123807 --resume work_dirs/faster-rcnn_r101_fpn_2x_coco/epoch_23.pth
大家可以根据自己的模型来替换路径和文件。
今天不努力,明天变垃圾!