程序地带

IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Pytho


报错原因分析:


train_loss += loss.data[0] 是pytorch0.3.1版本代码,在0.4-0.5版本的pytorch会出现警告,不会报错,但是0.5版本以上的pytorch就会报错,总的来说是版本更新问题.


解决方法:


#将原语句:


train_loss+=loss.data[0]


#修改为:


train_loss+=loss.item()


#bingo


 


承接Matlab、Python和C++的编程,机器学习、计算机视觉的理论实现及辅导,本科和硕士的均可,咸鱼交易,专业回答请走知乎,详谈请联系QQ号757160542,非诚勿扰。


 


版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/weixin_36670529/article/details/111246074

随机推荐

nginx的配置文件中变量是如何设计实现的

nginx的配置文件使用简单灵活,某些部分还具备脚本语言的特点,变量就是其中一个特色。本文将分析变量是如何设计实现的。0.什么是变量脚本语言都有变量这个东西,...

黑刀·秋水 阅读(746)

Linux-shell 字符子串获取

 前言这篇文章主要介绍shell脚本编程中的获取子串用法 一、常见的子串获取方法${parameter}返回变量$parameter的内容${#parameter}返回变量$parameter的内容长...

The best will coming 阅读(108)

idea打包项目,本地创建bat运行

1.idea执行maven命令package;2.在控制台可以看到本地文件的存放路径:3.进入到该目录下,可以看到2个文件:4.将这两个文件移动到别的地方&#...

学吧太深了 阅读(193)

PC个人隐私保护小方法

PC个人隐私保护小方法

前言近期爆出了腾讯读取用户浏览器浏览记录的消息。话不过说直接上图,懂的自然懂。网上也有详细的分析文章,不管它读取后用来做什么,在你不知情的情况下读取了你的浏览器浏览记录,你说气不气。虽然在整体大环境下...

kidicc 阅读(156)