训练YOLOv4，加载了权重报No module named 'mish_cuda'，未加载权重时报module 'torch.nn' has no attribute 'Mish'

iscyy / yoloair

🔥🔥🔥 专注于YOLOv5，YOLOv7、YOLOv8、YOLOv9改进模型，Support to improve backbone, neck, head, loss, IoU, NMS and other modules🚀

https://github.com/iscyy/yoloair

GNU General Public License v3.0

2.51k stars 430 forks source link

Closed jmlw8023 closed 1 year ago

jmlw8023 commented 1 year ago

iscyy commented 1 year ago

Install the corresponding mish_cuda