DataLoader 工作线程意外退出？

Question

我正在运行这篇Medium文章中看到的代码：https://medium.com/@bentanweihao/training-a-custom-object- detector-in-half-a-day-with-yolov8-5e1475fe201e

具体是实际训练部分：

from ultralytics import 
YOLO model = YOLO("yolov8n.pt") 
model.train(data="data.yaml", epochs=20) 
metrics = model.val()  

我收到了这个错误：

     25         def decorate_context(*args, **kwargs):
     26             with self.__class__():
...
--> 999                 raise RuntimeError('DataLoader worker (pid(s) {}) exited unexpectedly'.format(pids_str)) from e
   1000             if isinstance(e, queue.Empty):
   1001                 return (False, None)

RuntimeError: DataLoader worker (pid(s) 21812, 23384, 6272, 20544, 27792, 27784, 17928, 15588, 28072, 24112) exited unexpectedly

有人有建议吗？

Answer 1

我的解决方案是这样的：

model.train(data="data.yaml", epochs=20,workers=0) 

这是由于 Windows 线程造成的。