运行lwdetr_tiny_coco_eval.sh出现这个问题怎么解决

Atten4Vis / LW-DETR

This repository is an official implementation of the paper "LW-DETR: A Transformer Replacement to YOLO for Real-Time Detection".

Apache License 2.0

248 stars 16 forks source link

运行lwdetr_tiny_coco_eval.sh出现这个问题怎么解决 #20

Open beyond-xy opened 4 months ago

beyond-xy commented 4 months ago

File ".\LW-DETR-main\models\matcher.py", line 88, in forward cost_class = pos_cost_class[:, tgt_ids] - neg_cost_class[:, tgt_ids] RuntimeError: CUDA error: device-side assert triggered Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.