invalid index of a 0-dim tensor. Use `tensor.item()`

最新推荐文章于 2024-04-26 21:45:21 发布

乱搭巴士

最新推荐文章于 2024-04-26 21:45:21 发布

阅读量1.1k

点赞数 4

CC 4.0 BY-SA版权

分类专栏： torch

本文链接：https://blog.csdn.net/qq_45128278/article/details/114590278

torch 专栏收录该内容

14 篇文章

订阅专栏

使用1.7.1版本的torch框架运行代码时出现问题报错如下：
invalid index of a 0-dim tensor. Use tensor.item() in Python or tensor.item<T>() in C++ to convet

源代码是在1.4.0版本的torch下运行的。

改动如下:

'losses.update(loss.data[0], input.size(0))' '在torch0.4的版本使用此句，高一些的版本用下面的用法'
        losses.update(loss.item(), input.size(0))
        'top1.update(prec1[0], input.size(0))'
        top1.update(prec1.item(), input.size(0))