RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cuda:0!

Braddy
Fri, 18 Aug 2023 08:10:59 +0000

Thanks for your deep dive. I made it after specifying the blocks containing residual connections as the “non-split” ones.