专栏算法工具链转多batch模型时在常量tensor的node转换时遇到ShapeInferenceError

转多batch模型时在常量tensor的node转换时遇到ShapeInferenceError

已解决
青芒-阿白2025-06-09
68
2

环境：J6M，OE：v3.0.31

input_batch：2会出现下面的错误，input_batch改为1正常，报shape不匹配的错误：

错误node：

错误log：

模型里只有这两个node用到了常量tensor，都报了上面的错误，请问这个问题应该怎么解决？

算法工具链
征程6
评论1
0/1000
  • 青芒-阿白
    Lv.1
    所有reshape和resize的shape参数，dim0无法自动推导为input_batch，导致后续的shape发生冲突，应该怎么修改？
    2025-06-10
    0
    1
    • Huanghui回复青芒-阿白:

      你好，你已经分析的对了，就是因为shape参数常量化，导致动态修改 input_batch 时无法（自适应）修改导致的问题，要解决该问题可以通过一以下两个方法之一：

      1. 根据模型结构计算好shape参数中出去第一维度之外的其他维度，然后第一维度使用-1，这样就可以自适应了。

      2. 不把shape产量化，在代码中通过data.shape[0]获取首维数据，然后组合其他维度。

      建议使用方法1.

      2025-06-11
      0