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