pytorch-beginner-master 填坑學習筆記

04-Convolutional Neural Network 出現tensorflow的問題, 註釋掉LOGGER的代碼.

loss.data[0] 報錯, 這是版本問題, 將所有的data[0]替換成item()後成功. 

image = Variable(image, volatile=True) 報錯, 替換成 with torch.no_grad(): image = Variable(image) 後成功

05-Recurrent Neural Network裏出現backward報錯, 在訓練循環裏的第一行增加  model.train() 後成功.

 

 

 

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