Set correct device for torch.arange for DETA

IDEA-Research / detrex

detrex is a research platform for DETR-based object detection, segmentation, pose estimation and other visual recognition tasks.

https://detrex.readthedocs.io/en/latest/

Apache License 2.0

1.92k stars 202 forks source link

Closed thangngoc89 closed 11 months ago

thangngoc89 commented 11 months ago

I believe the code changes explain itself.

rentainhe commented 11 months ago

I believe the code changes explain itself.

Thank u so much for fixing this