Pytorch测试神经网络时出现 RuntimeError:的解决方案
作者：漂洋过海的油条 发布时间：2022-03-28 14:26:47
标签：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():
来源：https://blog.csdn.net/weixin_40533355/article/details/93158700
0
投稿猜你喜欢
- 中国，美国，英国3国时间js同步动态显示，对于做企业网站的朋友相信用的到，特别是做英文网站的朋友，加上这一段代码会给你的网站增色不少！本文j
python发邮件需要掌握两个模块的用法，smtplib和email，这俩模块是python自带的，只需import即可使用。smtplib- 前沿小补充例3.48 查询平均成绩大于等于80分的学生学号和平均成绩SELECT Sno,AVG(Grade)FROM SCWHERE AV
- 索引 经常要查询的语句，则给它建一个索引 表连接 select T_Oders as o join T_Customers as C on
- type,元类，类，对象简单描述1.type是python内建元类，新建的元类需要继承type2.元类用来创建类，类用来创建对象类的生成方式
- 1 用mysql客户端登入 2 选择数据库 mysql>use test 3 查询当前数据库有哪些存储过程 mysql>show
- 描述int函数可以将一个指定进制的数字型字符串或者十进制数字转化为整形。语法int(object, base)名称说明备注object一个数
- Python实现对变位词的判断，供大家参考，具体内容如下什么是变位词呢？即两个单词都是由相同的字母组成，而各自的字母顺序不同，譬如pytho
- pyinstaller打包问题简单介绍一下pyinstaller常用的参数：可选参数示例说明-Fpyinstaller -F demo.py
- 本文实例讲述了Python使用dict.fromkeys()快速生成一个字典。分享给大家供大家参考，具体如下：>>> re
- 第一步：首先定义一个视图函数，用于提供数据，实现每页显示数据个数，返回每页请求数据from django.shortcuts import
- 本文实例讲述了JS实现简单的抽奖转盘效果。分享给大家供大家参考，具体如下：<!DOCTYPE html><html lan
- 前言在我们抓取网页内容的时候，通常是抓取一整个页面的内容，而我们仅仅只是需要该网页中的部分内容，那该如何去提取呢？本章就带你学习xpath插
- 用了smarty,tp过后，也想了解了解其模板技术是怎么实现，于是写一个简单的模板类，大致就是读取模板文件->替换模板文件的内容-&g
- 工作时同事间几mb小文件的传输，一般使用QQ或者微信就足够了，但当传输文件几百MB或者几十G时，这种方法的效率就显得不足了。本篇就是简单说明
- 用途:将UTF-8编码汉字转换为GB2312码,兼容英文和数字版权:虽说是原创，其实也参考了別人的部分算法asp源代码：<％
- 看代码，tensor转numpy:a = torch.ones(2,2)b = a.numpy()c=np.array(a) #也可以转nu
- 介绍百度aip模块是用于实现百度云与用户接口，简单来说就是使用百度云所拥有的人工智能模块。模块使用pip install baidu-aip
- ubuntu18.04版本，python版本python2.7,python3.5,python3.6因为安装一些库会安装到python3.
- 错误提示： Warning: session_start() [function.session-start]: Cannot send s
