在进行模型测试时报错:
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