知识经验分享——YOLOv5-6.0训练出错及解决方法（RuntimeError）

>>>深度学习Tricks，第一时间送达<<<

目录

一、问题bug

二、解决方法

步骤1：打开utils/loss.py文件

步骤2：找到 for i in range(self.nl) 函数（Ctrl+F），作以下替换：

步骤3：找到# Append部分（Ctrl+F），作以下替换：

步骤4：顺利运行train.py文件

---

一、问题bug

不知道小伙伴们在训练YOLOv5-6.0或者其他版本有没有遇到以下问题：

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

二、解决方法

步骤1：打开utils/loss.py文件

步骤2：找到 for i in range(self.nl) 函数（Ctrl+F），作以下替换：

替换代码：

anchors, shape = self.anchors[i], p[i].shape

替换之后：

步骤3：找到# Append部分（Ctrl+F），作以下替换：

替换代码：

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

替换之后：

步骤4：顺利运行train.py文件

参考链接：https://blog.csdn.net/Thebest_jack/article/details/125649451?ops_request_misc=&request_id=&biz_id=102&spm=1018.2226.3001.4187

关于算法改进及论文投稿可关注并留言博主的CSDN

>>>一起交流！互相学习！共同进步！<<<