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