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

 

 

print('Epoch[{}/{}], loss:{:.6f}'.format(epoch+1,num_epoch,loss.data[0]))

将loss.data[0] 改为loss.item()