Pytorch 調試錯誤記錄

Pytorch 調試錯誤記錄 加載模型錯誤，加載模型參數時，關鍵字不匹配，每個都多了「module」 解決方法：給自己的net再套一層torch.nn.DataParallel()就行，或者在你訓練的時候把這句話去掉，這句話是在你有多卡情況下用的，如果你只有一張卡可以直接去掉。

>>阅读原文<<