问题:
模型训练好,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