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

在进行模型测试时报错：

Missing key(s) in state_dict: xxxxxxxxxx

Unexpected key(s) in state_dict:xxxxxxxxxx

 

报错原因：

在模型训练时有加上：【可以加速训练速度】

model = nn.DataParallel(model)

#cudnn.benchmark = True

但是在模型测试推断时，在模型参数被加载到模型前没有加这句话，故报出上面的错误。

 

解决：

在模型参数被加载到模型前加下面的语句：

model = nn.DataParallel(model)

#cudnn.benchmark = True