×

签到

分享到微信

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

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

error: &#039onnx.Unsqueeze&#039 op requires a single operand 已完结 胡汉三2022-11-25 16:04:53 回复 6 查看 技术答疑 使用求助 经验交流
error: &#039onnx.Unsqueeze&#039 op requires a single operand
分享到:

使用onnx_build在解析bert_ _buggingface.onnx文件时,报错如下:


2022-11-25 07:56:09.364228: INFO: /usr/local/neuware/samples/magicmind/mm_build/main.cc:13]

==================== Parameter Information

--precision<Str>              : force_float16

--calibration<Bool>           : false

--file_list<Vec<Str>>         :

--calibration_data_path<Str>  :

--random_calib_range<Vec<Float>>:

--batch_size<Vec<Int>>        :

--input_dims<Vec<Vec<Int>>>   : 64,256 64,256 64,256

--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>>          : mtp_372

--plugin<Vec<Str>>            :

--magicmind_model<Str>        : bert_ _huggingface_bs64_seq256.model

--build_config<Str>           : config.json

--toolchain_path<Str>         : /usr/local/gcc-linaro-6.2.1-2016.11-x86_64_aarch64-linux-gnu/

--rgb2bgr<Bool>               : false

--onnx<Str>                   : /data/models/test_models/models/onnx/bert_squad/bert_ _huggingface.onnx

MagicMind: 13631488

error: 'onnx.Unsqueeze' op requires a single operand

2022-11-25 07:56:10.978465: ERROR:  magicmind/translation/onnx/module_loader.cc:45] Failed to verify onnx mlir module

error: 'onnx.Unsqueeze' op requires a single operand

error: 'onnx.Unsqueeze' op requires a single operand



所运行命令如下:

./onnx_build --onnx /data/models/test_models/models/onnx/bert_squad/bert_ _huggingface.onnx --precision force_float16 --input_dims 4,128 4,128 4,128 --calibration false --mlu_arch mtp_372 --build_config config.json --magicmind_model bert_ _huggingface_bs4_seq128.model


===

请问是否是因为版本导致的op不支持?

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