yolov5：我可以在已经完成自定义训练的 model 上添加更多 epoch 吗？

Question

我正在尝试使用 30,000 张图像进行图像检测 model。 我刚刚尝试了 5 个 epoch，大约需要 5 个小时的训练。

问题是，如果我想在同一个 model 上简单地添加更多时期，我该怎么做？ 有没有办法可以加载经过自定义训练的 model？

由于需要这么长时间，我希望我可以从已经完成训练的 5 个 epoch 继续训练。 我正在使用 colab，我在训练期间经历了几次失去联系，所以我害怕从训练开始就给出更多的 epoch。

Answer 1

是的，您可以使用标准的 yolo 训练命令，并在 --weights 参数中提供您之前训练 model 的路径。

!python train.py --img 960 --batch 20 --epochs 10 --data dataset.yaml --weights your_model.pt --cache --exist-ok