Toybrick

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

作者: geoergecyr 时间: 2020-11-17 11:48
标题: KeyError: 'aten::norm'
torchscript格式的pytorch模型，转rknn报错，具体内容如下图所示:

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

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

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

作者: geoergecyr 时间: 2020-11-17 14:26

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，请问这个时候应该怎么办呢？

作者: jefferyzhang 时间: 2020-11-17 14:33

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

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

作者: zhaomr 时间: 2020-11-23 17:28

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

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

作者: jefferyzhang 时间: 2020-11-23 20:15

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

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

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

作者: rakidis 时间: 2020-12-24 15:11
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

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