found at least two devices: cuda:0 and cpu

[user52Real]

Hello,

first of all thank you for the project.
I have an issue running your python script fit_single_img.py and Im getting this error :

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
Are there any commands to specify the device?

Thank you in advance.

[ascust]

Could you be able to locate the error more accurately? Like which line caused the error. A general suggestion is that you print out the device of the variables before this specific operation and convert the device accordingly.