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

[katariaak579]

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

[PLUS-WAVE]

using python3 infer.py --deca DECA --rome data --device cuda