KeyError: 'aten::grid_sampler'

[复制链接] · 发表于 2020-6-17 10:00:22

本帖最后由 heizhima 于 2020-6-17 10:09 编辑

torchscript格式的pytorch模型，转rknn报错，具体内容如下:

是不是rknn，目前不支持这个算子？

只看该作者 · 发表于 2020-6-17 10:28:43

torch版本是1.2.0？
模型保存前有eval？

只看该作者 · 发表于 2020-6-17 19:43:34

leok 发表于 2020-6-17 10:28
torch版本是1.2.0？
模型保存前有eval？

torch版本是1.2.0,
模型保存前有eval.

只看该作者 · 发表于 2020-6-17 19:59:37

leok 发表于 2020-6-17 10:28
torch版本是1.2.0？
模型保存前有eval？

您好，我后面试了其他模型，从pytorch到rknn，也是这个KeyError，好像pow也是有问题。

只看该作者 · 发表于 2020-6-18 18:44:55

你是用rknn toolkit 1.3.2测试的么？不是的话请在1.3.2上测试完我才能报op给NPU部门。

只看该作者 · 发表于 2020-6-19 15:25:43

jefferyzhang 发表于 2020-6-18 18:44
你是用rknn toolkit 1.3.2测试的么？不是的话请在1.3.2上测试完我才能报op给NPU部门。
...

您好，我rknn toolkit 版本从1.3.0更新到1.3.2进行测试也是一样的问题，详见如下截图：

只看该作者 · 发表于 2020-6-19 16:52:51

好的，我已经报给NPU部门了