On Dual GPU systems you can run into the following error:
!!! Exception during processing !!! Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cuda:0! (when checking argument for argument tensors in method wrapper_CUDA_cat)
This PR addresses that issue by ensuring that the tensors are all loaded on the same device.
On Dual GPU systems you can run into the following error:
This PR addresses that issue by ensuring that the tensors are all loaded on the same device.