RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0! (when checking argument for argument index in method wrapper_CUDA__index_select)
用时:0.6 sec.
突然提示这个了 有没有大佬帮看看怎么解决