IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python
pytorch报错:IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python一般是torch版本的不同造成的。解决:将代码中的loss_return.data[0] 改成loss_return.datanorm += params.norm(2)....
原创
2019-03-28 17:56:39 ·
1035 阅读 ·
0 评论