Closed AXYZdong closed 6 months ago
add .to(device) in line 209, 214, 366, 368. If not, may have the error RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
.to(device)
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
add
.to(device)in line 209, 214, 366, 368. If not, may have the errorRuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!