1. 程式人生 > >Pytorch中常見的報錯解決方案

Pytorch中常見的報錯解決方案

本文用於記錄所在pytorch所遇到過的執行時錯誤,持續更新

  1. 變數所在裝置(CPU,GPU)不一致問題

RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

可能原因:現在假設程式碼要在GPU上執行,並且你已經進行my_model.to(device)操作了。注意只有my_model中的屬性(也就是self.開頭的變數)才會進行.to(device)。如果出現這個錯誤,可能是有的中間變數需要手動再顯式地.to(device)一下~