解决yolov5-5.0 RuntimeError: result type Float can’t be cast to the desired output type long int

Yolov5-5.0训练的时候，之前就遇到过问题如下

 貌似当时解决了，从网上重新下的代码，本地train.py的时候一点问题都没有，可以正常python train.py（windows环境下），但是一模一样的代码拷贝到ubuntu搭建好的环境下运行的时候就会报如上错误。

解决办法：

（1）改变loss.py178行内容

（2）改变loss.py211行内容，改为下图大红箭头指向的内容。

indices.append((b, a, gj.clamp_(0, shape[2] - 1), gi.clamp_(0, shape[3] - 1)))  # image, anchor, grid

 如上两步更改完成之后，可以美美的训练啦。。。

python train.py

 

ref：(30条消息) 一步真实解决RuntimeError: result type Float can‘t be cast to the desired output type long int_蓝胖胖▸的博客-CSDN博客