Thank you for the reply again, but it doesn’t match what I’ve seen in practice. When I run the second piece of code in the original question and run the model net, I get results that I do not expect, implying that the code failed to move the state dict correctly to cuda:0. Is this expected behaviour, and if not should I try to make a minimum example and flag an issue?