专栏算法工具链[求助]QAT流程中使用hbdk4.export工具导出bc模型时伪数据出错

[求助]QAT流程中使用hbdk4.export工具导出bc模型时伪数据出错

已解决
溺水小坦克2025-01-04
53
2
芯片型号:J6M
OE版本:3.0.22
我在使用自己的模型做QAT的导出qat.bc和定点bc模型时,仿照bevformer例程中的导出脚本export_hbir.py,构造伪输入数据并调用hbdk4.export进行模型导出,但是会报错,看起来是显示输入数据不匹配,报错:

但我是仿照bevformer的输入伪数据方式进行伪数据的构造,不知道为什么在我的模型上这种格式的输入会报错,请问这可能是什么原因导致的?
bevformer例子的伪数据:

我自己模型的伪数据:
我的模型定义的输入为:




算法工具链
征程6
评论1
0/1000
  • Huanghui
    Lv.5

    你好,目前从你提供的信息没有看出具体原因,我尝试通过OE附带的示例显示类似的错误时可以看到类似“INFO Exporting hbir with input [{'queries_rebatch_grid': “的日志内容,你的应该也是有的吧,请配合做一下事情:

    1. 提交完整的日志。

    2. 提供模型导出是所用的配置文件和和bc文件。

    2025-01-07
    0
    1
    • 溺水小坦克回复Huanghui:

      您好,经过排查,发现是我模型定义输入数据时没有按照dict的方式传入,修改后没有报错,可以导出bc模型文件。

      2025-01-08
      0