Toybrick

标题: KeyError: 'aten::grid_sampler' [打印本页]

作者: heizhima 时间: 2020-6-17 10:00
标题: KeyError: 'aten::grid_sampler'
本帖最后由 heizhima 于 2020-6-17 10:09 编辑

torchscript格式的pytorch模型，转rknn报错，具体内容如下:
[attach]1194[/attach]
是不是rknn，目前不支持这个算子？

作者: leok 时间: 2020-6-17 10:28
torch版本是1.2.0？
模型保存前有eval？

作者: heizhima 时间: 2020-6-17 19:43

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

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

作者: heizhima 时间: 2020-6-17 19:59

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

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

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

作者: heizhima 时间: 2020-6-19 15:25

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

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

作者: jefferyzhang 时间: 2020-6-19 16:52
好的，我已经报给NPU部门了

欢迎光临 Toybrick (https://t.rock-chips.com/)