专栏算法工具链多输入hybrid模型QAT后无法使用hb_mapper正确转换

多输入hybrid模型QAT后无法使用hb_mapper正确转换

ArthurThomas2023-08-19
21
4

用户您好,请详细描述您所遇到的问题,这会帮助我们快速定位问题~

1.芯片型号:X3
2.天工开物开发包OpenExplorer版本:openexplorer/ai_toolchain_ubuntu_20_xj3_gpu v2.6.2b
3.问题定位:模型转换
4.问题具体描述
按照https://developer.horizon.ai/forumDetail/168628789079880786的答复,使用QAT训练异构多输入模型后,得到符合预期的精度。

按照如下代码得到onnx模型

但是在进行hb_mapper编译时报错

下图为hb_mapper编译选项

下图为model_calibrated_model.onnx中包含HzDequantize_303的部分结构

算法工具链
评论2
0/1000
  • 颜值即正义
    Lv.2

    您好,请问可以提供一下onnx模型嘛,便于我们复现和解决问题

    2023-08-21
    0
    2
    • ArthurThomas回复颜值即正义:
      是上文中qat后的qat.onnx
      2023-08-21
      0
    • 颜值即正义回复ArthurThomas:

      是的

      2023-08-21
      0
  • 颜值即正义
    Lv.2
    你好,此问题已在新版本的工具链中fix,可以通过安装以下链接中的whl包解决:https://pan.horizon.ai/index.php/s/AjLYX9ZmswbQqWC
    2023-08-22
    0
    0