RuntimeError: Error(s) in loading state_dict for ***: Missing key(s) in state_dict:

問題:

分析:加載模型參數時,關鍵字不匹配,每個都多了“module”

解決:

只要給自己的net再套一層torch.nn.DataParallel(),就行。

發佈了35 篇原創文章 · 獲贊 14 · 訪問量 4萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章