Pytorch加載模型時報錯

加載模型時出現Unexpected key(s) in state_dict錯誤 報錯截圖如下： 反覆排查問題沒發現爲何如此，查看pytorch中文文檔發現保存和加載模型方法都完全正確，模型保存和加載代碼對比中文文檔截圖如下： 其中一個方法是在加載模型時添加參數strict=False，可以只保留鍵值相同的參數避免出錯，用法如下：model.load_state_dict(ckpt[‘state_

>>阅读原文<<