【Pytorch】Pytorch加载训练好的模型报错记录

基于Pytorch深度学习框架下，
使用2张GPU训练模型

加载训练好的模型时，报错如下：

密密麻麻的提示，看得我头皮发麻。。。。

经过一番搜索，看到网上也有人遇到和我一样的问题：
博主原文如下：
https://blog.csdn.net/shiwanghualuo/article/details/101756876

博主给出了原因，并且给出了解决方案，我没有按照博主给出的方案进行解决问题，我感觉应该有更加简单得方法去解决。于是在网上继续寻找…

原博文：https://blog.csdn.net/tsq292978891/article/details/83586935


说实话，以上两位博主的解决方案，都不是我想要的，指直到看到这位博主的文章：
https://www.cnblogs.com/wildkid1024/p/13025352.html

加载训练好的模型model时，再将其放入torch.nn.DataParallel()不就搞定了！

加了红框这行代码，不报错了，程序可以正常运行了。

特此记录！

参考：
https://blog.csdn.net/shiwanghualuo/article/details/101756876

https://blog.csdn.net/Cloris_Sue/article/details/109569231