pytorch报错tensor.item()——IndexError: invalid index of a 0-dim tensor.

IndexError: invalid index of a 0-dim tensor. Use `tensor.item()` in Python or `tensor.item<T>()` in C++ to convert a 0-dim tensor to a number

pytorch报错tensor.item()——IndexError: invalid index of a 0-dim tensor.

pytorch在训练时发生如上报错,有其他大佬解释是pytorch版本问题造成的.data[0]和.item()的问题,但我在创建虚拟环境,安装其它版本pytorch后依旧出现同样的错误,因此排除了pytorch版本问题。

出错位置如下:

pytorch报错tensor.item()——IndexError: invalid index of a 0-dim tensor.    咨询大佬后,明白了是在dataloader部分出现了问题,比如我batchsize设置为8,而最后一个batch图片数目不足8张,造成了维度不同问题,在dataloader中加入drop_last=True,丢弃图片数量不足8张的最后一个batch即解决。

如下:

pytorch报错tensor.item()——IndexError: invalid index of a 0-dim tensor.

 

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
上一篇 2022年5月27日 下午4:09
下一篇 2022年5月27日 下午4:12

相关推荐

本站注重文章个人版权,不会主动收集付费或者带有商业版权的文章,如果出现侵权情况只可能是作者后期更改了版权声明,如果出现这种情况请主动联系我们,我们看到会在第一时间删除!本站专注于人工智能高质量优质文章收集,方便各位学者快速找到学习资源,本站收集的文章都会附上文章出处,如果不愿意分享到本平台,我们会第一时间删除!