RuntimeError: CUDA error: device-side assert triggered

落日映苍穹つ 2022-10-09 02:47 146阅读 0赞

RuntimeError: CUDA error: device-side assert triggered

报错信息

  1. Traceback (most recent call last):
  2.   File "train.py", line 402, in <module>
  3.     train(hyp)
  4.   File "train.py", line 300, in train
  5.     results, maps, times = test.test(opt.data,
  6.   File "/home/zd/project/yolov5-deepsort/test.py", line 107, in test
  7.     loss += compute_loss([x.float() for x in train_out], targets, model)[1][:3]  # GIoU, obj, cls
  8.   File "/home/zd/project/yolov5-deepsort/utils/utils.py", line 427, in compute_loss
  9.     tobj[b, a, gj, gi] = (1.0 - model.gr) + model.gr * giou.detach().clamp(0).type(tobj.dtype)  # giou ratio
  10. RuntimeError: CUDA error: device-side assert triggered

解决办法

  1. 检查所有txt标签文件，是否有空文件。删掉空文件即可

欢迎大家交流学习，任何问题都可以留言

发表评论

表情：
评论列表 （有 0 条评论，146人围观）

还没有评论，来说两句吧...

相关阅读