IndexError: invalid index of a 0

提示 Use tensor.item() to convert a 0-dim tensor to a Python number

原语句：running_loss+=loss.data[0]

修改后：running_loss+=loss.item()