rome icon indicating copy to clipboard operation
rome copied to clipboard

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

Open katariaak579 opened this issue 1 year ago • 0 comments

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

katariaak579 avatar Feb 02 '24 13:02 katariaak579