search

AlanLi1997 / slim-neck-by-gsconv

Pytorch implementation of the 'Slim-neck by GSConv: a lightweight-design for real-time detector architectures'
https://link.springer.com/article/10.1007/s11554-024-01436-6
GNU General Public License v3.0
195 stars 12 forks source link

RuntimeError: result type Float can't be cast to the desired output type long int #28

Closed Amuleter closed 10 months ago

Amuleter commented 1 year ago

I faced error: result type Float can't be cast to the desired output type long int when train

solved by edit loss.py as below: https://github.com/ultralytics/yolov5/pull/8067/files

AlanLi1997 commented 10 months ago

I faced error: result type Float can't be cast to the desired output type long int when train

solved by edit loss.py as below: https://github.com/ultralytics/yolov5/pull/8067/files

Thanks for your feedback and I will fix this soon.