专栏算法工具链petr算法用hat工具进行编译时报错

petr算法用hat工具进行编译时报错

已解决
默认768322024-06-18
56
4

在编译compile_model时出现the model has 1 inputs, but given 2 sources types.,请问一下,编译时的inputs是来自哪里,是配置文件里的

def get_deploy_input(): 函数吗,问题是现在的函数里是一样的类型

def get_deploy_input():

  inputs = {

    "img": torch.randn((6, 3, 320, 640)),

    "pos_embed": torch.randn((1, 256,60, 20)),



    }

  return inputs


算法工具链
评论2
0/1000
  • Huanghui
    Lv.5

    收到,我们分析一下,请提供问题背景:1, 所使用的工具链版本; 2. 原始SAMPLE就有问题,还是在原始SAMPLE的基础上进行过修改; 3. 报错时的原始日志保存到文件中提供一下。

    2024-06-19
    0
    2
    • 默认76832回复Huanghui:

      docker 是horizon_j5_open_explorer_v1.1.68-py38_20231014的,其实我主要问题是想知道,这个配置的输入为什么不支持int型,就改了 petr原始的输入,其他没变

      2024-06-19
      0
    • Huanghui回复默认76832:

      该部分input的数据是在float2qat过程中使用的,该阶段使用的依然是浮点数据进行训练和推理的,所以不支持int型。

      2024-08-07
      0
  • Huanghui
    Lv.5

    客户您好,鉴于长时间未收到你的答复,我们暂将该问题做关闭处理,如您仍有相同问题,请重新开帖子进行提问,祝您工作愉快!

    2024-08-15
    0
    0