【成都AICC】ATC转换时设置动态shape,调用报错:动态尺寸注册失败,请检查模型配置
收藏回复举报
【成都AICC】ATC转换时设置动态shape,调用报错:动态尺寸注册失败,请检查模型配置
t('forum.solved') 已解决
发表于2024-04-18 11:46:12
0 查看

1【客户名称】:四川省寰宇众恒科技有限公司

2【设备型号】:Atlas 800 推理服务器(型号:3000)

3【运行环境】:

●CANN 版本 :社区版8.0.RC1.alpha003

●onnx 版本: 1.12.0

●Python 版本 :3.10.13

●操作系统版本 : 统信服务器操作系统V20(1020e)

4【问题现象描述&截图&日志】:

问题现象描述:根据官方介绍文档中的介绍,atc工具支持模型的动态分辨转换,但是将模型转换出来以后,转换以后模型的文件名应该为:image_preprocess.om,但实际模型的文件名称变为了image_preprocess_linux_aarch64.om,并且使用aclmdlGetInputIndexByName接口获取不到动态输入的index,得到的是一个很大的值20402880;然后在使用aclmdlSetInputDynamicDims接口注册初始化输入的分辨率维度,就会注册失败,返回值为100000。我去看了gitee sample库下的提问,发现有类似的问题,但最后都没有找到解决方案。

转换的指令为:

atc --model=image_preprocess.onnx --framework=5 --output=image_preprocess --soc_version=Ascend310P3 --input_shape="X:1,640~2560,640~2560,3;sub_x:3;mul_x:3" --input_format=ND

推理过程截图:

cke_1958.png

atc转换过程截图:

cke_2881.png

cke_3857.png

还发现转换后的om文件的大小存在异常,2.3konnx文件转换出了52Mom文件

我要发帖子