深度学习避坑提示：RuntimeError: Error(s) in loading state_dict for Model: Missing key(s) in state_dict

错误提示：

使用FCN，Unet，Deeplab训练完成的网络权重，加载模型的时候报错：

RuntimeError: Error(s) in loading state_dict for Model: Missing key(s) in state_dict....

训练环境：

CPU:Intel E5
GPU: 3090*2
Pytorch1.10

解决方式：

解决不匹配问题
方法1（无效）：字典不能严格匹配，在模型加载的时候使用如下代码添加False，解决不匹配问题

net.load_state_dict(t.load(ckpt_path),False)

注意：该方法解开严格匹配，能够忽略报错，但是用此方法加载的模型因为参数不匹配，在实际分割效果出现问题，慎用！

方法2:网络相同，唯一不同的是在训练过程中使用了nn.DataParallel()方式调用两张显卡进行训练，所以在加载模型权重之前，将模型同样适用DataParallel包装，可以解决上面的报错问题。

net = nn.DataParallel(net)
net = net.to(device)