错误提示：
使用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)
403
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
