problems
wzj3315182 opened this issue · 0 comments
wzj3315182 commented
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