1. 程式人生 > 其它 >IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python

IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python

技術標籤:經典目標檢測演算法執行錯誤ssd

IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python
在這裡插入圖片描述
train_loss += loss.data[0] 是pytorch0.3.1版本程式碼
解決方法:

#將原語句:

loss_l.data[0], loss_c.data[0],

loss_l.data[0], loss_c.data[0],

#修改為:

loss_l.item(), loss_c.item(),

loss_l.item(), loss_c.
item(),