onnx 转为 om 模型,输出丢失,onnx有7个输出,om模型只有5个输出
收藏回复举报
onnx 转为 om 模型,输出丢失,onnx有7个输出,om模型只有5个输出
t('forum.solved') 已解决
发表于2024-07-09 16:05:32
0 查看

我的onnx模型原来有7个输出,使用atc模型转换之后只有5个输出。

原来onnx的模型推理输出结构如下:

转化为om模型推理输出结构如下:

我的atc转换指令如下:

atc --model=models.onnx --framework=5 --output=models --soc_version=Ascend310B4

以下是我的模型结构图:

请问如何才能获取另外的两个输出(output_4, output_5)?

我要发帖子