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():
(资源库 www.zyku.net)
原文链接:https://blog.csdn.net/weixin_40533355/article/details/93158700
栏 目:Python教程
下一篇:python 装饰器的使用与要点
本文标题:Pytorch测试神经网络时出现 RuntimeError:的解决方案
本文地址:https://www.zyku.net/python/9903.html
您可能感兴趣的文章
- 02-11pytorch交叉熵损失函数的weight参数的使用
- 02-11pytorch 如何把图像数据集进行划分成train,test和val
- 02-11pytorch 6 batch_train 批训练操作
- 02-11pytorch固定BN层参数的操作
- 02-11pytorch 如何实现HWC转CHW
- 02-10pytorch 使用半精度模型部署的操作
- 02-10pytorch 中nn.Dropout的使用说明
- 02-10浅谈pytorch中的dropout的概率p
- 02-10基于PyTorch实现一个简单的CNN图像分类器
- 02-10pytorch中.to(device) 和.cuda()的区别说明
- 02-10Pytorch 中net.train 和 net.eval的使用说明
- 02-10Pytorch 如何训练网络时调整学习率
- 02-10pytorch model.cuda()花费时间很长的解决
- 02-10Pytorch GPU内存占用很高,但是利用率很低如何解决
- 02-09PyTorch 如何自动计算梯度
- 02-09pytorch 实现计算 kl散度 F.kl_div()
- 02-09pytorch中LN(LayerNorm)及Relu和其变相的输出操作
- 02-09pytorch 实现多个Dataloader同时训练
- 02-09解决pytorch trainloader遇到的多进程问题
- 02-09Pytorch使用shuffle打乱数据的操作
最近更新
阅读排行
猜你喜欢
- 03-24荣耀v40开启开发者选项方法
- 09-27央视频全运会闭幕式直播预约步骤分享
- 01-11yuvino美食分享-yuvino美食分享应用软
- 06-12nginx去除目录执行PHP的权限
- 03-31pytorch训练神经网络爆内存的解决方案
- 08-15小米mix4怎么开启负一屏
- 02-19Discuz QQ互联登录提时示错误代码1054
- 01-19猫猫抓娃娃-猫猫抓娃娃应用软件功能介
- 01-12A Design Kit-A Design Kit应用软件功
- 01-06小米体脂秤2数据在哪看