encounter1997 / DE-DETRs

Official Implementation of DE-DETR and DELA-DETR in "Towards Data-Efficient Detection Transformers"
Apache License 2.0
79 stars 8 forks source link

Training on my own dataset reports errors #3

Open xielaoban755 opened 11 months ago

xielaoban755 commented 11 months ago

After modifying the num_classes, it trained properly but reported an error in validation:

assert (boxes1[:, 2:] >= boxes1[:, :2]).all() RuntimeError: CUDA error: device-side assert triggered.

If the CoCo category number is still used no error occurs