Thanks! Yeah, I’ll make sure to use the same device.
Just a quick follow up though, what do the id of the parameters depend on?
model_1.cpu()
model_2.cpu()

model_2.gpu()
model_1.gpu()

Would the id of the parameters change in this case? Just trying to avoid possible mistakes.