RuntimeError: Error(s) in loading state_dict for ..:Missing key(s) in state_dict: …Unexpected key...
原因:预训练权重层数的键值与新构建的模型中的权重层数名称不吻合,Checkpoint里面的模型是在双卡上训练的,保存的key前面都多一个module.解决:model=torch.nn.DataParallel(model,device_ids=[0,1]).cuda()torch.nn.DataParallel是一种能够将数据分散到多张显卡上从而加快模型训练的方法。它的原理是首先在指定的每张显卡