专栏算法工具链.hbm可否自定义输入输出名称?

.hbm可否自定义输入输出名称?

已解决
HandsomeChen2024-03-13
71
7

1. 芯片型号：J5

2. 天工开物开发包 OpenExplorer 版本： J5_OE_1.1.62

3. 问题定位：模型转换

4. 问题具体描述：.hbm的输出名字能否自行指定？

.hbm输入输出的名字能否自定义？类似'arg[0]['img']'或者'backbone_conv0'这样的名字对于多输出的网络部署不是很方便，不能快速地将输出一个一个对应。请问能否通过compile_config等方式自定义输入输出的名称？

算法工具链
征程5
评论6
0/1000
  • GOYU\XU
    Lv.2

    模型输入节点的名称，应与模型文件中的名称一致，否则编译时会报错！

    2024-03-14
    0
    0
  • GOYU\XU
    Lv.2

    你可以在转onnx模型时指定模型输入输出名称

    2024-03-14
    0
    0
  • HandsomeChen
    Lv.1

    感谢您的回复，但是我转换的模型是.hbm，应该不需要生成onnx模型，onnx模型对hbm模型的input_name/output_name也没有影响。请问我这样的理解对吗？


    2024-03-14
    0
    0
  • GOYU\XU
    Lv.2

    模型如果已经转换为hbm，输出名字就没法改变，据我所知，量化感知训练过程中是预留了导出onnx模型的接口的

    2024-03-14
    0
    0
  • HandsomeChen
    Lv.1

    感谢回复。还想请教一下：按照我对模型转换的浅显理解，模型转换为板上运行模型一共有两种方式。

    1. 模型-----ONNX------.bin

    2. 模型--------------------.hbm

    如果是这样的话，即使我改变了ONNX的input/output_name，也没办法改变hbm模型中的输入输出名字。请问这样的理解对吗？

    2024-03-14
    0
    0
  • GOYU\XU
    Lv.2

    模型训练后转onnx,然后通过onnx转bin/hbm,如果模型转onnx过程中已经更改了输出名，那后面onnx转bin/hbm就沿用此名

    2024-03-14
    0
    1
    • 北海回复GOYU\XU:

      同问题，我走的是QAT流程，不需要经过Onnx模型，是直接torch模型转hbm模型，如何通过 config等方式自定义输入输出的名称

      2025-03-11
      0