Pytorch测试神经网络时出现 RuntimeError:的解决方案
更新时间：2021年05月28日 09:58:44 作者：漂洋过海的油条
这篇文章主要介绍了Pytorch测试神经网络时出现 RuntimeError:的解决方案，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
Pytorch测试神经网络时出现“RuntimeError: Error(s) in loading state_dict for Net”
解决方法：
load_state_dict(torch.load('net.pth')
在前，增加
model = nn.DataParallel(model)
就可以了。
比如
net = NET()
net.cuda()
net = nn.DataParallel(net)
net.load_state_dict(torch.load('net.pth')
补充：解决RuntimeError: Error(s) in loading state_dict for XXXX
在运行代码时遇到了这个错误，显示错误对应代码中的state_dict，找到对应的语句
修改前：
net.load_state_dict(torch.load(model_para_path))
修改后：
net.load_state_dict(torch.load(model_para_path),False)
我遇到的问题得到解决。经查阅资料，我理解的原因是state_dict的四个参数之一:_module 可以用来判断模型当前运行环境与之前是否相同，在默认情况下是True（纯属个人理解，勿喷）
补充：使用Pytorch训练模型出现RuntimeError: CUDA out of memory错误解决
训练：
由于GPU显存资源有限，训练输入的batchsize不能过大，过大会导致out of memory错误。
解决方案：
将batchsize减小，甚至是为1
测试时出现此问题解决方案：
在测试代码之前使用 with torch.no_grad():
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
pytorch geometric的GNN、GCN的节点分类方式
这篇文章主要介绍了pytorch geometric的GNN、GCN的节点分类方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-12-12
Python中的字典是无序类型,没有自己的排序方法,下面这篇文章主要给大家介绍了关于Python根据字典值对字典进行排序的三种方法,文中通过实例代码介绍的非常详细,需要的朋友可以参考下2022-09-09
Python基于matplotlib画箱体图检验异常值操作示例【附xls数据文件下载】
这篇文章主要介绍了Python基于matplotlib画箱体图检验异常值操作,涉及Python针对xls格式数据文件的读取、matplotlib图形绘制等相关操作技巧,并附带xls数据文件供读者下载参考,需要的朋友可以参考下2019-01-01
Python Opencv中用compareHist函数进行直方图比较对比图片
这篇文章主要介绍了Python Opencv中用compareHist函数进行直方图比较进行对比图片，本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2020-04-04
这篇文章主要介绍了python中如何设置list步长，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-05-05
这篇文章主要介绍了Python使用pandas将表格数据进行处理，文章围绕主题展开详细的内容介绍，具有一定的参考价值，感兴趣的小伙伴可以参考一下2022-08-08
这篇文章主要介绍如何利用 Python 实现多任务进程，正在执行的程序，由程序、数据和进程控制块组成，是正在执行的程序，程序的一次执行过程，是资源调度的基本单位。下面就来详细介绍改内容,需要的朋友可以参考一下2021-10-10
Python数据分析:手把手教你用Pandas生成可视化图表的教程
今天小编就为大家分享一篇Python数据分析:手把手教你用Pandas生成可视化图表的教程，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-12-12
今天小编就为大家分享一篇python3.6使用urllib完成下载的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-12-12
这篇文章主要介绍了python中的变量命名规则详情，变量名可以包括字母、数字、下划线，但是数字不能做为开头，变量用的好或不好，和代码质量有着非常重要的联系，合理的使用变量，可以让你的代码可读性更高并且更加简洁,下面相关内容吧需要的小伙伴可以参考一下2022-03-03
最新评论