Pytorch如何加载训在多GPU上训练好的模型?

请各位大神指点, 我在训练网络的时候用到了多GPU, 然后储存参数用了 "model.module.state_dict()", 我现在想用单GPU去测试模型需要加载参数, 这个时候可以用 "model.state_dict()" 来读取多GPU训练的模型参数吗?

多谢

 

不知道你这个问题是否已经解决, 如果还没有解决的话:

• 你可以看下这个问题的回答https://ask.csdn.net/questions/7428796
• 我还给你找了一篇非常好的博客，你可以看看是否有帮助，链接：pytorch将模型加载到GPU上
• 同时，你还可以查看手册：pytorch 保存和加载模型 中的内容

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 以帮助更多的人 ^-^