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
即可
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python
最新推荐文章于 2024-04-26 21:45:21 发布
本文解决了因torch版本不同导致的PyTorch报错问题，具体表现为IndexError异常。通过调整代码中loss_return.data[0]及norm+=params.norm(2).data[0]两处，替换成loss_return.data及norm+=params.norm(2).data即可解决。
扫一扫
1226
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
