51dev.com IT技术开发者社区

51dev.com 技术开发者社区

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

ymnets阅读(504)2020-03-15 收藏0次评论

在进行模型测试时报错:

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

以上就是pytorch错误解决:Missing key(s) in state_dict: Unexpected key(s) in state_dict:的全部内容。