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

应用开发 算法工具链
,
1

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

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。

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

2

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

3

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

4

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

5

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

mod=horizon.nn.quantized.FloatFunctional()

mod.matmul()

后再进行check

6

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

7

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