search

ZJU-lishuang / yolov5_prune

yolov5 prune，Support V2, V3, V4 and V6 versions of yolov5
Apache License 2.0
559 stars 137 forks source link

楼主，请问遇到RuntimeError: CUDA error: an illegal memory access was encountered怎么解决？ #52

Open Shiso-Q opened 3 years ago

Shiso-Q commented 3 years ago

4 楼主，请问遇到RuntimeError: CUDA error: an illegal memory access was encountered怎么解决？ 我后来将i = torchvision.ops.boxes.nms(boxes, scores, iou_thres)改为 i = non_max_suppressionv5(boxes,scores,iou_thres) 又出现了元组超出，请问这要怎么解决呢？ 2

ZJU-lishuang commented 3 years ago

检查类别数是否正确。 这个需要调试解决或者找个会调试代码的人。