Why the code is taking much time on GPU than CPU?

[zobeirraisi]

I run the code on my local GPU and CPU:
for CPU it takes 18 sec
but for GPU it takes 59 sec?
How it can be solved?

[acheman]

I run the code on my local GPU and CPU: for CPU it takes 18 sec but for GPU it takes 59 sec? How it can be solved?

Hello, I wonder if this code base is used to convert DETR into a detection box that can detect rotation? I modified it in the DETR code according to this code, what else do I need to modify so that I can detect the rotation detection frame (xc, yc, w, h, angle), thank you very much.

[craymichael]

This is in part due to all of the .to(device) calls in box_ops.py. They can be changed to e.g. torch.zeros(..., device=device) calls for each reference respectively