Is it possible to load a model which is stored with model.module.state_dict() but load with model.state_dict()

vision
1

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!

2

Yes, you should be able to load the state_dict via model.load_state_dict().
Are you seeing any issues using this approach?