pytorch錯誤：Missing key(s) in state_dict、Unexpected key(s) in state_dict解決

版權聲明：本文為博主原創文章，歡迎轉載，並請注明出處。聯系方式：[e-mail]

在模型訓練時加上：

model = nn.DataParallel(model)
cudnn.benchmark = True
可以加速訓練速度。
但是需要注意的是，訓練后保存的模型參數在被加載到模型前，需要對模型加上：

model = nn.DataParallel(model)
cudnn.benchmark = True

否則加載時會出現錯誤：
RuntimeError: Error(s) in loading state_dict for ResNet:
    Missing key(s) in state_dict: xxxxxxxx
　　Unexpected key(s) in state_dict: xxxxxxxxxx