yolov5继续训练模型

运行环境:Ubuntu18.04 + anaconda虚拟环境

1.断点续传

对于上次训练时意外中断的情况

将patser中第9个参数resume,设置为default=True

或运行:

python train.py --resume True

2.在上次训练结果的基础上继续开始训练

将patser中第1个参数weights,设置为上次运行完

~/yolov5/runs/train/exp/weights/best.pt (根据实际情况进行修改)

同样可以在运行时进行修改,如:

python train.py --weights runs/train/exp/weights/best.pt

你可能感兴趣的:(ubuntu)