专栏算法工具链QAT量化模型编译hbm文件过程中出现问题

QAT量化模型编译hbm文件过程中出现问题

已解决
默认383362025-05-25
53
2

您好,我在自己的数据集上参考oe包中提供的kitti数据集上pointpillars模型示例,用QAT量化pointpillars模型,在最后将pth转为hbm的过程中,当我input_source=["pyramid"]时出现如下报错:

我的config.py文件中的deploy_inputs设置为:

并非报错中提到的2 elements。我看compile_perf.py中就是从 deploy_inputs 读取的数据:deploy_inputs = cfg.deploy_inputs,请问这个报错的原因是什么?
算法工具链
征程5
评论2
0/1000
  • 费小财
    Lv.5

    pyramid "arg0[points][0]_torch_native" 检查下这个参数,看上去参数没对齐的

    2025-05-27
    0
    0
  • Huanghui
    Lv.5

    如果这是工具链中的示例模型,不应该有错误才对,你有没有做个修改啥呢?

    2025-05-29
    0
    0