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.data
norm += params.norm(2)....
原创
2019-03-28 17:56:39 ·
1072 阅读 ·
0 评论