专栏算法工具链使用工具量化时,报错,提示元素类型不匹配

使用工具量化时,报错,提示元素类型不匹配

已解决
ZQQ2025-09-05
41
5

2025-09-05 21:12:22,032 ERROR *** ERROR-OCCUR-DURING {horizon_nn.build_onnx} ***, error message: There is an ERROR during shape inference, error message: [ShapeInferenceError] (op_type:Pow, node name: /ReduceL2_Pow): [TypeInferenceError] Inferred elem type differs from existing elem type: (DOUBLE) vs (FLOAT). The error model has been saved as shape_inference_fail.onnx

提示元素类型不匹配,这种如何解决

算法工具链
征程5
评论3
0/1000
  • Jade-self
    Lv.5
    从报错看，是 ONNX 的 Pow 节点推理时，推理出的 dtype 是 DOUBLE（float64），但已有 dtype 是 FLOAT（float32），两者不一致导致 shape inference 失败 。
    排查建议：
    1. onnx模型本身使用onnxruntime能否正常推理？
    2. 使用pytorch2.x版本导出onnx，并在代码对应位置进行pow中数据类型强制都转换为 float32
    2025-09-06
    0
    2
    • ZQQ回复Jade-self:
      使用pytorch导出的onnx正常，且没有pow算子，但是使用OE包转换的时候，就莫名多出个pow算子，xiaoku_org.svg
      2025-09-08
      0
    • Huanghui回复ZQQ:

      你好这个问题解决了吗

      2025-09-16
      0
  • DR_KAN
    Lv.4

    可以确认下pytorch代码，这块数据有没有float64的，给他to一下转成float32

    2025-09-28
    0
    0
  • 费小财
    Lv.5

    强转有试过吗，还有没有对应的报错的？

    2025-09-30
    0
    0