在使用PyTorch进行预测时，遇到了RuntimeError，错误信息显示在加载CE_Net_的state_dict时缺少了MobileNetV3_encoder的相关权重。问题源于训练模型保存和加载的不匹配。解决方案未给出。
Pytorch预测时，加载权重遇到这种问题：
RuntimeError: Error(s) in loading state_dict for CE_Net_:
Missing key(s) in state_dict: “MobileNetV3_encoder.conv0.0.weight”, “MobileNetV3_encoder.conv0.1.weight”, “MobileNetV3_encoder.conv0.1.bias”, “MobileNetV3_encoder.conv0.1.running_mean”, “MobileNetV3_encoder.conv0.1.running_var”, “MobileNetV3_encoder.layer1.pw.0.weight”, “MobileNetV3_encoder.layer1.pw.1.weight”, “MobileNetV3_encoder.layer1.pw.1.bias”, “MobileNetV3_encoder.layer1.pw.1.running_mean”…
提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档
RuntimeError: Error(s) in loading state_dict for CE_Net_:
最低0.47元/天 解锁文章
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
