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():
以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了Python小红书旋转验证码识别实战教程,本文通过示例代码给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友参考下吧2023-08-08
下面小编就为大家分享一篇Python和Java进行DES加密和解密的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-01-01
这篇文章主要介绍了Python爬虫入门教程01：豆瓣Top电影爬取的方法，本文通过实例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2021-01-01
Task主要作用是帮助event_loop调度,如果Task发现它打包的协程在await一个对象(执行一个io操作),则Task会挂起该协程的执行,把控制权交回event_loop,让他去运行其他Task,当该对象运行完毕,Task将打包的协程恢复运行,本文介绍Python asyncio使用小结,感兴趣的朋友一起看看吧2024-02-02
在本篇文章里小编给大家整理了一篇关于python匿名函数的实例用法，有兴趣的朋友们可以学习参考下。2021-03-03
django生产环境搭建(uWSGI+django+nginx+python+MySQL)
本文主要介绍了django生产环境搭建，主要包括uWSGI+django+nginx+python+MySQL，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2021-08-08
TorchVision Transforms API目标检测实例语义分割视频类
这篇文章主要为大家介绍了TorchVision Transforms API大升级，支持目标检测、实例/语义分割及视频类任务示例详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-11-11
这篇文章主要为大家详细介绍了Python+Pyqt实现简单GUI电子时钟，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-07-07
这篇文章主要介绍了OpenCV HSV颜色识别及HSV基本颜色分量范围，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2019-03-03
浅谈keras中的Merge层(实现层的相加、相减、相乘实例)
这篇文章主要介绍了浅谈keras中的Merge层(实现层的相加、相减、相乘实例)，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-05-05
最新评论