Hello guys,
I want to ask a question, I have trained a model with two gpus and stored this model with model.module.state_dict(), now I want to load this model in one gpu, can I directly load this trained model with model.state_dict()?
Thanks in advance!