RuntimeError: Error(s) in loading state_dict for SSD: Missing key(s) in解決辦法

Anne332發表於2020-10-28

模型驗證的時候執行出現RuntimeError: Error(s) in loading state_dict for SSD:Missing key(s) in state_dict: 解決辦法|

net.load_state_dict(torch.load(args.trained_model))

改為

net.load_state_dict(torch.load(args.trained_model), False)

即可

主要是記錄一下自己平時遇到的問題,和大家分享一下
如有侵犯,請聯絡我

點個贊支援一下吧

相關文章