mindir模型无法在c++推理,算子缺少

问题:

模型训练好,checkpoint格式可以推理,可以转为mindir格式。但是c++ mindspore推理报错BroadcastTo缺失无法转为om模型推理。

[ERROR] GE_ADPT(20311,test_mindspore):2021-03-16-17:13:28.490.632 [mindspore/ccsrc/transform/graph_ir/convert.cc:102] FindAdapter] Can't find OpAdapter for BroadcastTo

[ERROR] ME(20311,test_mindspore):2021-03-16-17:13:28.490.757 [mindspore/ccsrc/cxx_api/model/model_converter_utils/multi_process.cc:110] ParentProcess] Catch parent process runtime error: mindspore/ccsrc/transform/graph_ir/convert.cc:102 FindAdapter] Can't find OpAdapter for BroadcastTo

[ERROR] ME(20311,test_mindspore):2021-03-16-17:13:29.589.857 [mindspore/ccsrc/cxx_api/model/acl/model_converter.cc:217] LoadMindIR] Convert MindIR model to OM model failed

[ERROR] ME(20311,test_mindspore):2021-03-16-17:13:29.589.912 [mindspore/ccsrc/cxx_api/model/acl/acl_model.cc:54] Build] Load MindIR failed.

Build model failed.

解决方案:

报错信息为onnx导出不支持broadcast算子。当前MindSpore导出onnx格式只支持resnet系列网络,后续会补充完善的。查看MindSpore的算子支持情况:https://www.mindspore.cn/doc/programming_guide/zh-CN/master/operator_list.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值