I'm running into this problem and it shows up like this。
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)
This is a problem that I can't seem to make it through my own
I'm running into this problem and it shows up like this。
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)
This is a problem that I can't seem to make it through my own