UserWarning: invalid index of a 0-dim tensor.
UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use tensor.item() to convert a 0-dim tensor to a Python number
.format(epoch+1, num_epochs, loss.data[0]))
将loss.data[0] 改为loss.item()