YOLOv8 Bug 及解决方案汇总 【2024.1.24更新】【环境安装】【训练 & 断点续训】OMPError / KeyError_m = getattr(torch.nn, m[3:]) if 'nn.' in m else gl

1. KeyError 问题

Traceback (most recent call last):
File “D:\Anaconda\Scripts\yolo-script.py”, line 33, in
sys.exit(load_entry_point(‘ultralytics==8.0.83’, ‘console_scripts’, ‘yolo’)())
self.model, self.save = parse_model(deepcopy(self.yaml), ch=ch, verbose=verbose) # model, savelist
File “D:\Anaconda\lib\site-packages\ultralytics-8.0.83-py3.8.egg\ultralytics\nn\tasks.py”, line 465, in parse_model
m = getattr(torch.nn, m[3:]) if ‘nn.’ in m else globals()[m] # get module
KeyError: ‘space_to_depth’
Sentry is attempting to send 2 pending error messages
Waiting up to 2 seconds
Press Ctrl-Break to quit

项目场景：

我发现目前很多同学在使用YOLOv8添加了自己的模块后都会出现这个报错。

问题描述:

比如我首先在