问题: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+/

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐