Toybrick

KeyError: 'aten::norm'

geoergecyr

中级会员

积分
211
楼主
发表于 2020-11-17 11:48:08    查看: 7606|回复: 6 | [复制链接]    打印 | 显示全部楼层
torchscript格式的pytorch模型,转rknn报错,具体内容如下图所示:

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


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

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

geoergecyr

中级会员

积分
211
沙发
 楼主| 发表于 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,请问这个时候应该怎么办呢?
回复

使用道具 举报

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

本版积分规则

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


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