File "/usr/local/lib/python3.8/dist-packages/ultralytics/nn/tasks.py", line 467, in parse_model m = getattr(torch.nn, m[3:]) if 'nn.' in m else globals()[m] # get module KeyError: 'CBAM'
时间: 2023-10-21 15:05:15 浏览: 542
/usr/lib/x86_64-linux-gnu/libopencv_videostab.so.2.4.8-附件资源
这错误的原因是你在代码中尝试使用了 CBAM 模块,但是该模块并不存在于 `torch.nn` 中。可能是你使用了一个自定义的模块或者第三方库中的模块。你需要确保该模块已经正确安装,并且可以被 Python 正确导入。如果你确定该模块已经正确安装,但是依然出现这个错误,那么你可以尝试在代码中手动导入该模块,例如:
```
from my_custom_module import CBAM
```
或者尝试使用其他模块代替 CBAM。
阅读全文