'Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!'

XinzeLee / RotateObjectDetection

This repository is based on Ultralytics/yolov5, with adjustments to enable rotate prediction boxes.

113 stars 22 forks source link

Open Zivid99 opened 2 years ago

Zivid99 commented 2 years ago

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!

DongChengdongHangZhou commented 2 years ago

I meet the same bug, how to deal with this problem?

Jxzzzzt commented 10 months ago

have you saved the problem? i met the same bug