1 系统环境
硬件环境(Ascend/GPU/CPU): Ascend
MindSpore版本: mindspore=2.6.0
执行模式(PyNative/ Graph): 不限
Python版本: Python=3.9
操作系统平台: Linux
2 报错信息
2.1 问题描述
如果换成ASin算子,还是不行,如果用数学公式替换,误差太大,有没有别的等效办法?
2.2 报错信息
Traceback (most recent call last):
File “/mnt/c/Users/33635/Desktop/DWPCNNFusion/onnx.py”, line 41, in
main()
File “/mnt/c/Users/33635/Desktop/DWPCNNFusion/onnx.py”, line 37, in main
export(net, input_ir, input_vis, file_name=‘fusion_model.onnx’, file_format=‘ONNX’)
File “/home/coke/.local/lib/python3.10/site-packages/mindspore/train/serialization.py”, line 2071, in export
_export(net, file_name, file_format, *inputs, **kwargs)
File “/home/coke/.local/lib/python3.10/site-packages/mindspore/train/serialization.py”, line 2126, in _export
_save_onnx(net, file_name, *inputs, **kwargs)
File “/home/coke/.local/lib/python3.10/site-packages/mindspore/train/serialization.py”, line 2183, in _save_onnx
onnx_stream = _executor._get_func_graph_proto(net, graph_id)
File “/home/coke/.local/lib/python3.10/site-packages/mindspore/common/api.py”, line 2266, in _get_func_graph_proto
return self._graph_executor.get_func_graph_proto(exec_id, ir_type, incremental)
RuntimeError: Can not find key Asin in convert map. Exporting Asin operator is not yet supported.
3 根因分析
如果是为了使用MindSpore Lite进行部署,可以直接使用mindspore训练脚本导出Mindir,mindspore支持导出mindir模型应该更方便,可以直接通过MindSpore Lite加载并推理。
4 解决方案
如果是想部署到昇腾的开发板的话,可以直接通过mindspore训练的脚本导出mindir模型,然后通过mindspore lite的转换工具,转换成mindspore lite格式的mindir(converter转换工具的使用方式可以参考mindspore lite官网教程),然后通过mindspore lite昇腾后端版本包进行部署。