测试模型是出现：RuntimeErrorError(s) in loading state_dict for MIMO-UNet

测试模型是出现：RuntimeError:Error(s) in loading state_dict for MIMO-UNet

在测试已经训练好的模型时，一直出现RuntimeError的错误，结合错误提示应该时模型加载时的错误。

在网上查了资料

1.加载使用模型时和训练模型时的环境不一致:

2.训练时使用了分布式训练方案nn.DataParallel()，在测试时没有没有将模型使用nn.DataParallel包起来.

解决方案：

在我的eval.py文件中，（这个文件有测试加载模型的代码，具体文件视情况而定）

将

model.load_state_dict(state_dict['model'])

改为：

model.load_state_dict(state_dict['model'],False)

采取这个方案能将模型成功加载出来，（load_state_dict方法参数的官方说明 strict 参数默认是true，他的含义是 是否严格要求state_dict中的键与该模块的键返回的键匹配）

但测试时明显感觉这个模型的指标过低，经过验证这样会影响模型加载的效果。

在我的项目中，我的环境是没有变化的，造成这个问题的原因是模型与网络结构不匹配，这个时候就得重新训练模型或者将网络结构修改成与当前模型一直才可。