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).data[0]改为norm += params.norm(2).data
即可
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).data[0]改为norm += params.norm(2).data
即可