Pytorch的onnx模型转离线om模型失败
收藏回复举报
Pytorch的onnx模型转离线om模型失败
t('forum.solved') 已解决
发表于2025-02-15 11:58:23
0 查看

我使用Pytorch 2.6最新版本中dynamo来将pth模型转成了onnx模型,使用如下命令行将其转换为离线om模型:

atc --model=./fft_model.onnx --framework=5 --output=./fft_model.om --output_type=FP32 --soc_version=xxxx --input_shape="input:1,16384,2" --input_format=ND --precision_mode_v2=origin

提示如下出错信息:

ATC start working now, please wait for a moment.
option model=./fft_model.onnx
option framework=5
option output=./fft_model.om
option output_type=FP32
option soc_version=Ascend310P3
option input_shape=input:1,16384,2
option input_format=ND
option precision_mode_v2=origin
...
ATC run failed, Please check the detail log, Try 'atc --help' for more information
E16005: [PID: 35688] 2025-02-15-11:52:44.357.444 The model has [2] [--domain_version] fields, but only one is allowed.
        TraceBack (most recent call last):
        Model parse to graph failed, graph name:fft_model.om.[FUNC:ModelParseToGraph][FILE:onnx_parser.cc][LINE:919]
        ATC model parse ret fail.[FUNC:ParseGraph][FILE:omg.cc][LINE:774]

使用MagicONNX库修正后,运行如下命令:

atc --model=./fft_model.onnx --framework=5 --output=./fft_model.om --output_type=FP32 --soc_version=Ascend310P3 --input_shape="input:1,16384,2" --input_format=ND --precision_mode_v2=origin

提示如下出错信息:

ATC start working now, please wait for a moment.
option model=./fft_model_mag.onnx
option framework=5
option output=./fft_model.om
option output_type=FP32
option soc_version=Ascend310P3
option input_shape=input:1,16384,2
option input_format=ND
option precision_mode_v2=origin
...
ATC run failed, Please check the detail log, Try 'atc --help' for more information
E19010: [PID: 36056] 2025-02-15-11:55:07.024.243 No parser is registered for Op [node_Cast_1, optype [pkg.onnxscript.torch_lib.common::1::Cast]].
        Solution: Check the version of the installation package and reinstall the package. For details, see the operator specifications.
        TraceBack (most recent call last):
        No parser is registered for Op [node_Reshape_3, optype [pkg.onnxscript.torch_lib.common::1::Reshape]].
        No parser is registered for Op [node_Cast_5, optype [pkg.onnxscript.torch_lib.common::1::Cast]].
        No parser is registered for Op [node_Reshape_7, optype [pkg.onnxscript.torch_lib.common::1::Reshape]].
        No parser is registered for Op [node_Cast_8, optype [pkg.onnxscript.torch_lib.common::1::Cast]].
        No parser is registered for Op [node_Reshape_10, optype [pkg.onnxscript.torch_lib.common::1::Reshape]].
        No parser is registered for Op [node_Slice_12, optype [pkg.onnxscript.torch_lib.common::1::Slice]].
        No parser is registered for Op [node_Gather_13, optype [pkg.onnxscript.torch_lib.common::1::Gather]].
        No parser is registered for Op [node_Cast_14, optype [pkg.onnxscript.torch_lib.common::1::Cast]].
        No parser is registered for Op [node_Reshape_16, optype [pkg.onnxscript.torch_lib.common::1::Reshape]].
        No parser is registered for Op [node_Cast_17, optype [pkg.onnxscript.torch_lib.common::1::Cast]].
        No parser is registered for Op [node_Reshape_19, optype [pkg.onnxscript.torch_lib.common::1::Reshape]].
        No parser is registered for Op [node_Cast_20, optype [pkg.onnxscript.torch_lib.common::1::Cast]].
        No parser is registered for Op [node_Reshape_22, optype [pkg.onnxscript.torch_lib.common::1::Reshape]].
        No parser is registered for Op [node_Slice_24, optype [pkg.onnxscript.torch_lib.common::1::Slice]].
        No parser is registered for Op [node_Gather_26, optype [pkg.onnxscript.torch_lib.common::1::Gather]].
        No parser is registered for Op [node_Unsqueeze_28, optype [pkg.onnxscript.torch_lib.common::1::Unsqueeze]].
        No parser is registered for Op [node_Unsqueeze_30, optype [pkg.onnxscript.torch_lib.common::1::Unsqueeze]].
        No parser is registered for Op [node_DFT_31, optype [pkg.onnxscript.torch_lib.common::1::DFT]].
        No parser is registered for Op [node_Squeeze_32, optype [pkg.onnxscript.torch_lib.common::1::Squeeze]].
        No parser is registered for Op [node_Shape_33, optype [pkg.onnxscript.torch_lib.common::1::Shape]].
        No parser is registered for Op [node_Gather_35, optype [pkg.onnxscript.torch_lib.common::1::Gather]].
        No parser is registered for Op [node_ReduceProd_36, optype [pkg.onnxscript.torch_lib.common::1::ReduceProd]].
        No parser is registered for Op [node_CastLike_37, optype [pkg.onnxscript.torch_lib.common::1::CastLike]].
        No parser is registered for Op [node_Unsqueeze_38, optype [pkg.onnxscript.torch_lib.common::1::Unsqueeze]].
        No parser is registered for Op [node_Unsqueeze_39, optype [pkg.onnxscript.torch_lib.common::1::Unsqueeze]].
        No parser is registered for Op [node_DFT_40, optype [pkg.onnxscript.torch_lib.common::1::DFT]].
        No parser is registered for Op [node_Squeeze_41, optype [pkg.onnxscript.torch_lib.common::1::Squeeze]].
        No parser is registered for Op [node_Shape_42, optype [pkg.onnxscript.torch_lib.common::1::Shape]].
        No parser is registered for Op [node_Gather_43, optype [pkg.onnxscript.torch_lib.common::1::Gather]].
        No parser is registered for Op [node_ReduceProd_44, optype [pkg.onnxscript.torch_lib.common::1::ReduceProd]].
        No parser is registered for Op [node_CastLike_45, optype [pkg.onnxscript.torch_lib.common::1::CastLike]].
        Model parse to graph failed, graph name:fft_model.om.[FUNC:ModelParseToGraph][FILE:onnx_parser.cc][LINE:919]
        ATC model parse ret fail.[FUNC:ParseGraph][FILE:omg.cc][LINE:774]

我的模型是一个简单的做快速傅利叶变换的模型,所以需要使用最新版Pytorch中的dynamo模式转换才能成功,想转成华为离线模型,在华为Atlas 300I Pro上运行,请问怎么解决这个问题?

本帖最后由 匿名用户2025/02/17 17:37:48 编辑

我要发帖子