解决记录 RuntimeError: CUDA error: device-side assert triggered

在使用北京超级云服务器进行训练YOLO V5 目标检测网络时,出现RuntimeError: CUDA error: device-side assert triggered这个问题。解决方法是:这种错误往往是因为你的标签数据里出现了超出标签范围的值,比如说你要做200个目标种类的识别,那么你的标签范围应该是0,1,2…199.一般我们采用从0开始的计数方式,但是如果你的标签数据里出现了诸如-1或200这样不再此范围中的数据时,便会报此错误。解决办法就是确认你的标签的范围,以及你产生标签的方式,是不是从1开始计数的。
但是我的标签的范围,产生标签的方式,是不是从1开始计数都是正确的。经过很多次尝试,我发现images文件夹下的train文件夹里的要训练的图片没有0.jpg这个文件,我把0.jpg加到train文件夹中,并把对应的xml文件也放到labels文件夹下的train文件夹中。 再次训练,没有报错。

你可能感兴趣的:(深度学习,pytorch,目标检测)