Expected all tensors to be on the same device, but found at least two devices, cpu and cuda:0 error message
I’m working on a model that’s written to run on either GPU or CPU exclusively but since I’m working on a HPC cluster, I’m running both.
I’m working on a model that’s written to run on either GPU or CPU exclusively but since I’m working on a HPC cluster, I’m running both.