×

签到

分享到微信

打开微信,使用扫一扫进入页面后,点击右上角菜单,

点击“发送给朋友”或“分享到朋友圈”完成分享

onnx_build生成mm模型报错 已完结 Leodoe2024-07-14 11:12:29 回复 13 查看 技术答疑 使用求助 社区交流
onnx_build生成mm模型报错
分享到:

 我使用的是magicmind:1.7.0-x86_64-ubuntu20.04-py_3_7镜像

执行 ./onnx_build --onnx onnx/vit-b-16.txt.fp16.onnx 报错了,

2024-07-14 02:51:46.825073: INFO: /usr/local/neuware/samples/magicmind/mm_build/main.cc:20] 
==================== Parameter Information
--precision<Str>              : 
--calibration<Bool>           : false 
--rpc_server<Str>             : 
--calibration_algo<Str>       : linear 
--file_list<Vec<Str>>         : 
--calibration_data_path<Str>  : 
--random_calib_range<Vec<Float>>: 
--batch_size<Vec<Int>>        : 
--input_dims<Vec<Vec<Int>>>   : 
--input_layout<Vec<Str>>      : 
--input_dtypes<Vec<Str>>      : 
--dynamic_shape<Bool>         : true 
--vars<Vec<Vec<Float>>>       : 
--means<Vec<Vec<Float>>>      : 
--output_layout<Vec<Str>>     : 
--output_dtypes<Vec<Str>>     : 
--mlu_arch<Vec<Str>>          : 
--mlu_product_name<Vec<Str>>  : 
--plugin<Vec<Str>>            : 
--magicmind_model<Str>        : ./model 
--build_config<Str>           : 
--toolchain_path<Str>         : /usr/local/aarch64--glibc--stable-2020.08-1/ 
--rgb2bgr<Bool>               : false 
--cluster_num<Vec<Vec<Int>>>  : 
--remote_calibration_plugin<Str>: 
--onnx<Str>                   : onnx/vit-b-16.txt.fp16.onnx 
--external_data_dir<Str>      :  
MagicMind: 1.7.0
error: 'onnx.Unsqueeze' op requires a single operand
2024-07-14 02:51:48.309926: ERROR:  magicmind/translation/onnx/module_loader.cc:61] Failed to verify onnx mlir module
error: 'onnx.Unsqueeze' op requires a single operand
error: 'onnx.Unsqueeze' op requires a single operand
error: 'onnx.Unsqueeze' op requires a single operand
error: 'onnx.Unsqueeze' op requires a single operand
2024-07-14 02:51:48.312868: ERROR:  magicmind/cc/internal/mm_parser_core.h:57]  Internal: Onnx mlir graph is illegal
2024-07-14 02:51:48.319902: ERROR: /usr/local/neuware/samples/magicmind/mm_build/parser.cc:20] Internal: Onnx mlir graph is illegal
Aborted (core dumped)

请教一下这个问题怎么修复。

版权所有 © 2025 寒武纪 Cambricon.com 备案/许可证号:京ICP备17003415号-1
关闭