专栏算法工具链GridSamplePlugin不支持

GridSamplePlugin不支持

已解决
yy12342024-09-24
93
3

1. 芯片型号:J5

2. 天工开物开发包 OpenExplorer 版本: J5_OE_1.1.77

3. 问题定位:模型转换

4. 问题具体描述:在量化带grid_sample算子的transformer模型时,报错:

ERROR *** ERROR-OCCUR-DURING {runtime.runtime_model_generation} ***, error message: HorizonRT not support these cpu operators: GridSamplePlugin。

我们在其他模型有type为HzQuantizedGridSample是正常的,这个模型中有type为GridSamplePlugin的,报错也是说的不支持这个cpu算子,这两种有什么区别呢?

我们的GridSample都是同一个类实现的,为什么量化工具会识别成两种类型。

netron可视化的两种类型的GridSample也是一样的:

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

    收到,我们想分析一下,稍后回复!

    2024-10-17
    0
    0
  • Huanghui
    Lv.5
    你好,目前j5支持的onnx opset为10/11, 而grid_sample是opset16中的算子,所以目前无法对原生的onnx的grid_sample进行支持。然而,在当前比较流行的transformer模型中grid_sample又被广泛使用,地平线针对此种情况引入了地平线特有的grid_sample进行支持,从参数上与onnx的grid_sample保持一致,具体信息可以参考:地平线开发者社区 (horizon.cc)
    2024-10-17
    0
    0
  • Huanghui
    Lv.5

    客户您好,长时间未收到你的答复,相信问题已解。如对此尚存疑问欢迎新帖讨论,感谢您的参与!

    2024-10-23
    0
    0