pytorch自學踩坑記錄(一)

教程《深度學習框架pytorch快速開發與實踐》邢夢來編著
出版社 電子工業出版社
買的應該是正版書,但是書中太多錯誤了,甚至很多低級的排版錯誤和打印錯誤。對於現在出版社的質量把關感到無語。

原文打印loss

print(('epoch [%d/%d],step [%d/%d],loss: %.4f'%(epoch+1,num_epochs,i+1,
                                                                len(train_dataset)/batch_size,loss.data[0])))

不知道是不是api換了,報錯,應該更正如下

print(('epoch [%d/%d],step [%d/%d],loss: %.4f'%(epoch+1,num_epochs,i+1,
                                                                len(train_dataset)/batch_size,loss.item())))

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章