search

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.47k stars 428 forks source link

Update loss.py & fixed training issue #139

Open akashAD98 opened 1 year ago

akashAD98 commented 1 year ago

fixed issue while training model issue fixed https://github.com/iscyy/yoloair/issues/118 Training yields: RuntimeError: result type Float can't be cast to the desired output type long int

akashAD98 commented 1 year ago

@iscyy you guys accept pr?

iscyy commented 1 year ago

Thank you for your PR, but it has been modified here, #48

akashAD98 commented 1 year ago

@iscyy no you can run code on google collab & you will get this issue. i check it but still, I'm getting error, so I created pr