一个torch版本报错
生活随笔
收集整理的這篇文章主要介紹了
一个torch版本报错
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
lossVal += losstensor.data[0]
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number
解釋:如果是torch0.3.x版本的代碼在torch0.4.x版本環境下跑,會報出如上錯誤,應作如下修改
原代碼:lossVal += losstensor.data[0] 修改后:lossVal += losstensor.item()?
總結
以上是生活随笔為你收集整理的一个torch版本报错的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: Faster RCNN中的交替训练
- 下一篇: pytorch——torch.backe