My model includes two models, a pre-trained model A --> model B.
I have trained model A on GPU0.
I fixed the model A and train the model on GPU1.
But I got the following error:
RuntimeError: expected device cuda:0 but got device cuda:1
Can anyone help me? Thanks a lot.