NotImplementedError YOLO ультралитики ⇐ Python

[Anonymous]

При запуске обучения нейронной сети получаю такую ​​ошибку:

Код: Выделить всё

NotImplementedError                       Traceback (most recent call last)
Cell In[1], line 5
1 from ultralytics import YOLO
3 model = YOLO('yolov8m.pt')
----> 5 results = model.train(data='dataset_yolo8/data.yaml', epochs=100, imgsz=640) #, model='yolov8m.pt')

File c:\Users\Mugmazoid\AppData\Local\Programs\Python\Python312\Lib\site-packages\ultralytics\engine\model.py:601, in Model.train(self, trainer, **kwargs)
598             pass
600 self.trainer.hub_session = self.session  # attach optional HUB session
--> 601 self.trainer.train()
602 # Update model and cfg after training
603 if RANK in (-1, 0):

File c:\Users\Mugmazoid\AppData\Local\Programs\Python\Python312\Lib\site-packages\ultralytics\engine\trainer.py:208, in BaseTrainer.train(self)
205         ddp_cleanup(self, str(file))
207 else:
--> 208     self._do_train(world_size)

File c:\Users\Mugmazoid\AppData\Local\Programs\Python\Python312\Lib\site-packages\ultralytics\engine\trainer.py:322, in BaseTrainer._do_train(self, world_size)
320 if world_size > 1:
321     self._setup_ddp(world_size)
--> 322 self._setup_train(world_size)
324 nb = len(self.train_loader)  # number of batches
325 nw = max(round(self.args.warmup_epochs * nb), 100) if self.args.warmup_epochs > 0 else -1  # warmup iterations
...
PythonTLSSnapshot: registered at ..\aten\src\ATen\core\PythonFallbackKernel.cpp:162 [backend fallback]
FuncTorchDynamicLayerFrontMode: registered at ..\aten\src\ATen\functorch\DynamicLayer.cpp:494 [backend fallback]
PreDispatch: registered at ..\aten\src\ATen\core\PythonFallbackKernel.cpp:166 [backend fallback]
PythonDispatcher: registered at ..\aten\src\ATen\core\PythonFallbackKernel.cpp:158 [backend fallback]

Все пути к файлам верны. Вот весь код:

Код: Выделить всё

from ultralytics import YOLO

model = YOLO('yolov8m.pt')

results = model.train(data='dataset_yolo8/data.yaml', epochs=100, imgsz=640, batch = 40) #, model='yolov8m.pt')

В чем может быть проблема и как ее устранить?

Подробнее здесь: https://stackoverflow.com/questions/779 ... ltralitycs