使用ATC将onnx模型转为Om模型
收藏回复举报
使用ATC将onnx模型转为Om模型
t('forum.solved') 已解决
发表于2025-02-12 22:58:23
0 查看

目标:将onnx模型转为om模型,使用ACL在昇腾上推理。

步骤:

1、运行装换命令

atc --model=25-1-20-crosspoint_model_00051.onnx --framework=5 --input_shape="x.1:1,6,2048;input:20,3,224,224" --output=crosspoint_model_00051  --soc_version=Ascend310P3

日志:

cke_27307.png

结果:生成了一个带有*_linux_aarch64.om结尾的模型,和fusion_result.json文件。

cke_7358.png

问题:请求这个转换正常吗?我命令中并没有指定生成_linux_aarch64.om后缀。正常的模型不应该生成这个后缀。使用ACL推理,发现转换好的模型跟pytorch训练推理出来的模型,在数值上差异挺大的。

我要发帖子