IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number
pytorch 版本问题
Try to change
total_loss += loss_val.data[0]
loss_values = [v.data[0] for v in losses]
to
total_loss += loss_val.data
loss_values = [v.data for v in losses]