search

SamsungLabs / rome

Realistic mesh-based avatars. ECCV 2022
Other
431 stars 41 forks source link

Device error on running rome_inference.ipynb with GPU #26

Open mgp123 opened 1 year ago

mgp123 commented 1 year ago

Hi everyone, On running the rome_inference.ipynb on a colab the following error appears on the evaluate step. Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument mat2 in method wrapper_CUDA_bmm) Probably due to a missing .to(device) somewhere? Thanks