print(‘Epoch[{}/{}], loss:{:.6f}‘.format(epoch+1,num_epoch,loss.data[0]))
将loss.data[0] 改为loss.item()
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number
原文：https://www.cnblogs.com/huangxiaofu/p/11454168.html
