专栏算法工具链在运行model_checker.py检查模型能否板端编译运行的时候报错

在运行model_checker.py检查模型能否板端编译运行的时候报错

已解决
beyondaa2023-04-21
23
6

用户您好，请详细描述您所遇到的问题，这会帮助我们快速定位问题~

1.芯片型号：J5
2.天工开物开发包OpenExplorer版本：J5_OE_1.1.40
3.问题定位：模型转换
4.问题具体描述：python3 tools/model_checker.py
在torch.matmul(points, rot) 这个算子的时候报ValueError: ('unsupported node', %new_coord.1 : Tensor = aten::matmul。

请问这是什么导致的，该怎么解决呢？

算法工具链
征程5
评论4
0/1000
  • 颜值即正义
    Lv.2

    您好，请首先尝试把torch.matmul算子替换为:

    mod=horizon.nn.quantized.FloatFunctional()

    mod.matmul()

    后再进行check

    2023-04-23
    0
    1
    • beyondaa回复颜值即正义:

      您好，之前就已经替换成这个了，还是不行

      2023-04-24
      0
  • beyondaa
    Lv.1

    您好，之前就试过替换这个算子，一直不行

    2023-04-24
    0
    1
    • 颜值即正义回复beyondaa:

      从报错来看是matmul计算不支持在cpu上进行，请您确保在gpu上进行操作

      2023-04-24
      0
  • beyondaa
    Lv.1

    您好，之前就已经替换成这个了，还是不行

    2023-04-24
    0
    0
  • beyondaa
    Lv.1

    您好，之前已经替换这个算子了，还是不可以

    2023-04-24
    0
    0