测试模型是出现:RuntimeErrorError(s) in loading state_dict for MIMO-UNet
测试模型是出现:RuntimeError:Error(s)inloadingstate_dictforMIMO-UNet在测试已经训练好的模型时,一直出现RuntimeError的错误,结合错误提示应该时模型加载时的错误。在网上查了资料1.加载使用模型时和训练模型时的环境不一致:2.训练时使用了分布式训练方案nn.DataParallel(),在测试时没有没有将模型使用nn.DataParalle