Pytorch错误集锦

RuntimeError: Expected one of cpu, cuda, mkldnn, opengl, opencl, ideep, hip, msnpu device type at start of device string: 0

在python\lib\site-packages\torch\cuda_init_.py文件在这里插入代码片中的set_device方法修改

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

YAMLLoadWarning: calling yaml.load() without Loader=... is deprecated, as the default Loader is unsafe. Please read https://msg.pyyaml.org/load for full details.yaml_cfg = edict(yaml.load(f))

方法一：在config.py文件中修改cfg_from_file方法（见错误日志），
为yaml_cfg = edict(yaml.load(f), Loader=yaml.FullLoader)
方法二：更新yaml
pip install pyyaml==3.12
3.

assert语句：
断言：assert语句用于检测表达式是否为真，如果为假，引发AssertionError错误。
语法：assert expression [,args]
expression：表达式

 size mismatch for encoder.weight: copying a param with shape torch.Size

查了半天结果是用了gpu，这是因为在多gpu上训练的模型在保存时候在参数名前多加了一个“module.”前缀，加载的时候把这个前缀去掉就行了：

device = torch.device('cpu')
# device = torch.device('cuda:0')
state_dict = torch.load('xxx.pth', map_location=device)from collections import OrderedDict
state_dict_new = OrderedDict()
for k, v in state_dict.items():name = k[7:]  # 去掉 `module.`state_dict_new[name] = vmodel.load_state_dict(state_dict_new)model.eval()

Pytorch错误集锦相关推荐

在Hadoop 2.3上运行C++程序各种疑难杂症（Hadoop Pipes选择、错误集锦、Hadoop2.3编译等）
首记感觉Hadoop是一个坑,打着大数据最佳解决方案的旗帜到处坑害良民.记得以前看过一篇文章,说1TB以下的数据就不要用Hadoop了,体现不出太大的优势,有时候反而会成为累赘.因此Hadoop的使 ...
x264代码剖析（四）：vs2010编译x264错误集锦
x264代码剖析(四):vs2010编译x264错误集锦支持VC++平台的x264的最新版本是x264-20091006,接下来就以该版本为例分析编译运行x264过程中遇到的问题以及解决办法. 1. ...
使用Kubeadm来搭建k8s-v1.18.2（包含所有错误集锦分析）
k8s的搭建使用 Kubeadm来搭建master集群,⽬前所安装的版本是 v1.18.2 欢迎运维萌新大佬等进群,涵盖业务运维.应用运维.系统运维.网络运维.数据库运维.桌面运维.运维开发等,地区 ...
.net开发过程中，错误集锦
2007年7月10日开发过程中,错误集锦. 写这个东东的目的,主要是上班那不能够上网,回家后总结出来弄到日志本里面的.顺便鄙视下移动,对厂家这么刻薄,不能上网,不能带水进去喝,最气人的是,进出大楼, ...
错误集锦：在安装docker的时候，发现无法安装（没有可用软件包 docker-ce。错误：无须任何处理）
错误集锦在我使用aliyun在线源安装docker的时候,出现以下错误,无法安装docker. [root@m1zi~]#yum install -y docker-ce 已加载插件:fastest ...
PyTorch 错误 RuntimeError: invalid argument 5: k not in range for dimension at /pytorch/aten/src/THC/g
PyTorch 错误 RuntimeError: invalid argument 5: k not in range for dimension at /pytorch/aten/src/THC/g ...
51单片机实验软件proteus出现错误集锦
51单片机实验软件proteus出现错误集锦错误一:安装KELI和proteus后,点击仿真但是出现下面错误. 错误二:点击仿真后,源代码本来有程序后面变空白,或者出不来这个调试窗口. 错误一:安装 ...
pytorch 错误 AttributeError: ‘Tensor‘ object has no attribute ‘backword‘
pytorch 错误 AttributeError: 'Tensor' object has no attribute 'backword' 按错误说明是没有backword属性. 错误代码 loss ...
pytorch 错误 ImportError numpy.core.multiarray failed to import
pytorch 错误 ImportError: numpy.core.multiarray failed to import 前言我下载安装 Pytorch1.7.1 GPU版,但在测试是否支持GP ...

Pytorch错误集锦

Pytorch错误集锦相关推荐

最新文章

热门文章