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

SamsungLabs / rome

Realistic mesh-based avatars. ECCV 2022

Other

428 stars 41 forks source link

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

Open katariaak579 opened 7 months ago

katariaak579 commented 7 months ago

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument grid in method wrapper_CUDA__grid_sampler_2d) whilke running infer.py