torch.Tensor类型的数据loss和acc打印时 ,如果写成以下写法
print('batch_loss: '+str(loss.data)+'batch acc: '+str(acc.data))
则不仅会打印出loss和acc的值,还会打印出device信息和 tensor字样,如下:
如果仅想打印出数值,
torch.Tensor类型的数据loss和acc打印时 ,如果写成以下写法
print('batch_loss: '+str(loss.data)+'batch acc: '+str(acc.data))
则不仅会打印出loss和acc的值,还会打印出device信息和 tensor字样,如下:
如果仅想打印出数值,