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
投稿猜你喜欢
- MySQL Innodb 引擎优化，InnoDB给MySQL提供了具有提交，回滚和崩溃恢复能力的事务安全(ACID兼容)存储引擎。InnoD
本文实例讲述了Python3.5迭代器与生成器用法。分享给大家供大家参考，具体如下：1、列表生成式通过列表生成式可以直接创建一个列表。代码：- 代码如下: <％ dim fso,objFolder,objFiles dim filelist Set fso=Server.Cre
- 场景分析一般Linux系统默认自带两个版本的python,我按照的虚拟机系统自带的版本为python2.7和python3.2,但是由于个人
- break 语句Python break语句，就像在C语言中，打破了最小封闭for或while循环。break语句用来终止循环语句，即循环条
- 终于构建出了第一个神经网络，Keras真的很方便。之前不知道Keras这么方便，在构建神经网络的过程中绕了很多弯路，最开始学的TensorF
- 索引是提高数据查询最有效的方法，也是最难全面掌握的技术，因为正确的索引可能使效率提高10000倍，而无效的索引可能是浪费了数据库空间，甚至大
- Dreamweaver从出道以来，其简便易用而又功能强大的“行为”(Behavior)一直是大家所关注的热点之一。在Dreamweaver
- 一、引用计数基础知识每个php变量存在一个叫 zval 的变量容器中。一个 zval 变量容器，除了包含变量的类型和值，还包括两个字节的额外
- python代码 # -*- coding: utf-8 -*- import Image,ImageDraw,Imag
- re正则表达式模块还包括一些有用的操作正则表达式的函数。下面主要介绍compile函数。 定义： compile(pattern[,flag
- import sysfrom PyQt5 import QtWidgetsfrom PyQt5.QtWidgets import QMain
- 这篇文章主要介绍了Python遍历字典方式就实例详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友
- 前言大家都知道golang的defer关键字，它可以在函数返回前执行一些操作，最常用的就是打开一个资源（例如一个文件、数据库连接等）时就用d
- 某人使用如下过滤代码，发现经常被黑：n=request.form("username") p=request.form(
- 1.经典类与新式类在了解Python的类与类型前，需要对Python的经典类（classic classes）与新式类（new-style
- 使用type()查看数据的类型在Python中, 可以使用type()类型来查看数据的类型：>>> type(3)<
- 本文实例讲述了python 函数的缺省参数使用注意事项。分享给大家供大家参考，具体如下：python的函数支持4种形式的参数：分别是必选参数
- 前言这篇文章主要介绍了pyinstaller打包opencv和numpy程序运行错误解决,文中通过示例代码介绍的非常详细，对大家的学习或者工
- 问题背景目前的linux发行版上，有很多安装了两个版本的python。我的机器上默认的版本为python 2.x。且在使用easy_inst
