报错IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number
解决:train_loss+=loss.data[0] 修改为 train_loss+= loss.item()
报错IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number
解决:train_loss+=loss.data[0] 修改为 train_loss+= loss.item()