欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a python number

程序员文章站 2024-03-17 11:24:16
...

提示 Use tensor.item() to convert a 0-dim tensor to a Python number

原语句:running_loss+=loss.data[0]

修改后:running_loss+=loss.item()

报错原因:版本升级

pytorch代码从0.3版本升级版本后的错误