Hi,
So I did the training of my neural network, say on GPU #3 and the state dict is saved. Do I have to use GPU #3 every time when I want to load my state dict?
Also, will it work if I don’t put my model to cuda before loading the state dict into the model?
Thank you.