torch.Tensor類型的數據loss和acc打印時 ,如果寫成以下寫法
print('batch_loss: '+str(loss.data)+'batch acc: '+str(acc.data))
則不僅會打印出loss和acc的值,還會打印出device信息和 tensor字樣,如下:
如果僅想打印出數值,使得打印出的信息更加簡潔,則要用以下寫法
print('batch_loss: {:.3f} batch acc: {:.3f}'.format(loss.data, acc.data))
該寫法還可控制輸出格式,結果類似如下: