高版本pytorch出现IndexError: invalid index of a 0-dim tensor.问题解决办法
最近使用pytorch1.4版本训练网络,在运用loss.data()时,目的时为了让loss函数的值相加,但出现问题IndexError: invalid index of a 0-dim tensor.问题解决办法。
经查找,发现问题的原因
loss += loss.data[0] 是pytorch0.3.1版本代码,在0.4-0.5版本的pytorch会出现警告,不会报错,但是0.5版本以上的pytorch就会报错,总的来说是版本更新问题.
该问题的解决办法:
第一种:
如果不想更改代码,则需要安装低版本的pytorch。
但不推荐这种做法。因为低版本的在现在很多开源新算法中已经被抛弃了。
第二种:
将代码中的
loss+=loss.data[0]
修改为:
train_loss+=loss.item()
改完之后就会愉快的进行网络的训练了。
高版本pytorch出现IndexError: invalid index of a 0-dim tensor.问题解决办法相关推荐
- IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python
报错原因分析: train_loss += loss.data[0] 是旧版本的Pytorch版本代码,在Pytorch0.4-0.5中会出现警告,不会报错, 在Pytorch0.5版本以上会报错 I ...
- IndexError: invalid index of a 0-dim tensor. Use `tensor.item()` in Python or `tensor.item<T>()` in
使用python pytorch框架出现问题: IndexError: invalid index of a 0-dim tensor. Use tensor.item() in Python or ...
- 解决IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Pyth
解决IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Pyth ...
- 【mean teacher】IndexError: invalid index of a 0-dim tensor. Use `tensor.item()` in Python o的解决
仅作为记录,大佬请跳过. 即可. 参考:传送门
- 【问题记录】pytorch自定义数据集 No such file or directory, invalid index of a 0-dim
保存模型: : 保存整个神经网络的结构和模型参数 torch.save(mymodel, 'mymodel.pkl') 只保存神经网络的模型参数 torch.save(mymodel.state_di ...
- IndexError:invalid index of a 0-dim tensor
运行从githb上下载下来的人脸表情识别项目时,报出如下错误 Traceback (most recent call last):File "mainpro_CK+.py", li ...
- proe服务器高速缓存位置,一招搞定Proe低版本打开高版本的问题 | 我爱分享网
有时我们会在网上下载一些proe或者Creo模型,由于不知道这些模型具体版本,使用软件打开时可能会在信息栏提示"xxx不能检索",这是因为Proe和Creo的低版本在默认情况下是无 ...
- node高版本不支持win7系统
背景: 当不同项目所使用的node版本要求不同时,我们需要安装多版本的node,这时就需要使用node多版本管理工具 nvm了,安装地址: https://github.com/coreybutler ...
- 幸福的烦恼:显卡算力太高而pytorch版本太低不支持
NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch instal ...
最新文章
- 怎样知道pip install 可以安装包的哪些版本
- DOM0,DOM2,DOM3事件,事件基础知识入门
- MySql数据备份与恢复
- 实际工程里的长宽设定
- Leetcode 118. 杨辉三角 (每日一题 20210901)
- 白话Elasticsearch62-进阶篇之Highlighting高亮显示
- 聊聊引用和 ThreadLocal 那些事儿
- Elasticsearch Reference [6.7] » Modules » Network Settings
- HDU_3786 找出直系亲属- softbar
- python标准化输出到txt_3大利器推荐,帮你写出规范漂亮的python代码
- java使用python爬虫,如何使用 Python 爬虫爬取 Java 题库?
- php宠物管理系统的开题报告,宠物医院管理系统的设计与实现开题报告.doc
- 四管前级怎么去掉高低音音调_烧友自荐:2SK304四管前级制作难点浅析
- OpenGL第十讲——像素图
- H5,Audio音乐播放器(移动版)
- Nodejs之child_process
- 【偶像大师 白金星光】的【Variable Tone】技术大公开!偶像从哪里看都那么可爱,VA小组谈制作方针...
- 【历史上的今天】4 月 25 日:诺基亚与微软完成交易;Mobilinux 发布;长距离无线电传输之父出生
- Microsoft Windows Win32k本地提权漏洞分析
- 康拓展开与逆康拓展开
热门文章
- wxpython制作eml文件阅读器
- 探索三维世界:从Hello World开始的Three.js入门之旅
- JS正则对象(正则表达式)方法属性总结及表单验证例题
- Photoshop-人物牙齿的美白亮化处理
- 运行jar文件出现了Exception in thread “main java.lang.UnsupportedClassVersionError
- 联邦学习框架和数据隐私综述
- MySQL之根据经纬度查询多少公里范围内的数据
- Blocks(poj 1390) 动态规划 方盒游戏 (升维——三维)
- SQL进阶--EXISTS谓词的用法
- centos 下安装ati显卡驱动方法