- 博客(1)
- 资源 (1)
- 收藏
- 关注
原创 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 number 是你的torch版本的不同造成的。 解决:将loss.data[0] 改成loss.item() ...
2019-02-28 22:00:56 20730 14
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人