Closed HeinrichAD closed 2 years ago
This changes will pay attention to the device while root_mask creation.
root_mask
Fixes RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
This changes will pay attention to the device while
root_mask
creation.Fixes
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!