KeyError: 'aten::norm'

[复制链接] · 发表于 2020-11-17 11:48:08

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

请问目前是否支持这个算子？

转换环境：Ubuntu18.04，RKNN1.3.2，torch版本是1.2.0，模型保存前有eval。

只看该作者 · 发表于 2020-11-17 11:58:23

1. 换成最新的rknn-toolkit尝试
2. rknn-toolkit的文档里都有op支持列表，请自行查阅

只看该作者 · 发表于 2020-11-17 14:26:51

jefferyzhang 发表于 2020-11-17 11:58
1. 换成最新的rknn-toolkit尝试
2. rknn-toolkit的文档里都有op支持列表，请自行查阅 ...

已经查阅了1.4.0 rknn，https://github.com/rockchip-linu ... P_Support_V1.4.0.md
也是没有对这个算子的支持，而且1.4.0版本的在转换torch模型时会遇到ZeroDivisionError，所以我才使用的1.3.2，请问这个时候应该怎么办呢？

只看该作者 · 发表于 2020-11-17 14:33:13

geoergecyr 发表于 2020-11-17 14:26
已经查阅了1.4.0 rknn，https://github.com/rockchip-linux/rknn-toolkit/blob/master/doc/RKNN_OP_Suppo ...

进置顶帖官方qq群里提建议

只看该作者 · 发表于 2020-11-23 17:28:04

jefferyzhang 发表于 2020-11-17 11:58
1. 换成最新的rknn-toolkit尝试
2. rknn-toolkit的文档里都有op支持列表，请自行查阅 ...

rknn1.4.0不支持pytorch的expand_as()有没有什么其他办法，谢谢！

只看该作者 · 发表于 2020-11-23 20:15:04

zhaomr 发表于 2020-11-23 17:28
rknn1.4.0不支持pytorch的expand_as()有没有什么其他办法，谢谢！

进官方QQ群里讨论，我个人是没有办法的，我不是专职做这个的。
大体思路：

1. 修改模型实现。
2. 提交需求给NPU部门让他们下一版本支持这个op。
3. 换Tensorflow
4. 转ONNX再转RKNN（可尝试）

只看该作者 · 发表于 2020-12-24 15:11:16

In my experience, rockchip cannot handle "Aten" operations

It handles onnx better, but even with that it can have problems.

It handles tensorflow(old versions) and caffe well, but all other frameworks are like beta

KeyError: 'aten::norm'

本帖子中包含更多资源