RuntimeError in util.py : torch.cuda.Floattensor about yolo_v3_tutorial_from_scratch

[AvivSham]

I also encountered the same problem, I have tried to change a few things without success.
Help, please.

I think I found the source of the problem. When you use:
print(x_y_offset.device) print(prediction.device)

you get cuda:0 for x_y_offset and cpu for prediction!
I just don't know how to solve this. I'm using colab notebook.

[AvivSham]

OK, I got it! (works for me)
After prediction[:,:,4] = torch.sigmoid(prediction[:,:,4])
add: prediction = prediction.cuda()

without the added line prediction is calculated in the CPU thus the problem!

Please keep me posted if it solves your problem.

[siddharthkale97]

OK, I got it! (works for me)
After prediction[:,:,4] = torch.sigmoid(prediction[:,:,4])
add: prediction = prediction.cuda()

without the added line prediction is calculated in the CPU thus the problem!

Please keep me posted if it solves your problem.

I tried with adding prediction = prediction.cuda()
but I kept on getting the error.
But I aslo encountered a strange thing. While following the tutorial I got this problem, but strangely the problem was solved when I cloned the repo and then run the program on a test timage.
Anyway thanks for your help