打开微信,使用扫一扫进入页面后,点击右上角菜单,
点击“发送给朋友”或“分享到朋友圈”完成分享
使用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不支持?
热门帖子
精华帖子