pytorch错误：Missing key(s) in state_dict、Unexpected key(s) in state_dict解决

在模型训练时加上：

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

2022-12-23
2022-01-08
2021-07-14
2021-10-01
2021-06-25
2021-11-22
2021-05-16
2021-05-06