Toybrick

KeyError: 'aten::norm'

jefferyzhang

版主

积分
12953
楼主
发表于 2020-11-17 11:58:23 | 显示全部楼层
1. 换成最新的rknn-toolkit尝试
2. rknn-toolkit的文档里都有op支持列表,请自行查阅
回复

使用道具 举报

jefferyzhang

版主

积分
12953
沙发
发表于 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群里提建议
回复

使用道具 举报

jefferyzhang

版主

积分
12953
板凳
发表于 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(可尝试)
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表