问题:yolov8训练中断,不想从头开始训练;或者训练一定数量的epoch后,想继续训练
PyTorch 2.0.0
Python 3.8(ubuntu20.04)
Cuda 11.8
解决方法:
打开.\ultralytics-main\ultralytics\cfg\default.yaml,看到model一栏,这就是程序train、test模式下使用的模型
如果是中途中断,将这里修改为最新生成的模型,best.pt是在一组epoch(比如我这里设置的是10,也就是程序训练了10个epoch就会自动停下来)中训练出的最好的模型
如果是觉得训练得不够,想增强训练,那就先改model,再改epochs,看自己要再训练多少
温馨提示:不要的语句可以先注释掉,而不是直接删掉哈,之后可能会用
注释:快捷键Ctrl+/
取消注释:快捷键Ctrl+Shift+/