Toybrick

RKNN哪种架构下支持RNN网络呢?

qiaoqiang126

中级会员

积分
225
楼主
发表于 2019-2-21 14:03:02    查看: 30640|回复: 13 | [复制链接]    打印 | 只看该作者
如题,在onnx 和tensorflow下测试都不支持RNN或者LSTM

回复

使用道具 举报

jefferyzhang

版主

积分
13578
沙发
发表于 2019-2-21 18:33:28 | 只看该作者
支不支持跟网络的结构是无关的,只要里头op支持,理论上是都支持的。如果出现rknn转换失败的情况,请把转换出错的verbose log发给我们分析。
回复

使用道具 举报

qiaoqiang126

中级会员

积分
225
板凳
 楼主| 发表于 2019-2-27 10:19:09 | 只看该作者
关于RNN很奇怪,tensorflow里面把Tensor ,unstack小于6个的时候,RNN是可以转成功的。但是UNstack大于6个的时候转换就失败了,UNstack,concat,rnn应该都是多于6个就没法转了。请问有解决办法吗?
附件中是verbose
回复

使用道具 举报

qiaoqiang126

中级会员

积分
225
地板
 楼主| 发表于 2019-2-27 10:30:01 | 只看该作者
能否公布一下onnx ,tensorflow ,tflite,caffe支持的OP呢  这样开发起来容易很多。
回复

使用道具 举报

qiaoqiang126

中级会员

积分
225
5#
 楼主| 发表于 2019-2-27 10:31:44 | 只看该作者
jefferyzhang 发表于 2019-2-21 18:33
支不支持跟网络的结构是无关的,只要里头op支持,理论上是都支持的。如果出现rknn转换失败的情况,请把转换 ...

能不能公开一下onnx ,tensorflow , caffe ,tflite支持的OP  。这样开发起来容易很多。
回复

使用道具 举报

jefferyzhang

版主

积分
13578
6#
发表于 2019-2-27 11:01:30 | 只看该作者
本帖最后由 jefferyzhang 于 2019-2-27 11:05 编辑
qiaoqiang126 发表于 2019-2-27 10:31
能不能公开一下onnx ,tensorflow , caffe ,tflite支持的OP  。这样开发起来容易很多。 ...

我估计rknntool 1.0发布时候应该会一起发布op列表的。
目前op都不存在支不支持的说法,只是那边npu驱动和op需要码代码去支持,所以目前所有不支持的op只能算是一个bug,我们会报给相关部门去支持的。而且目前op数量一直在增长,跟着所有开源架构一起增长有很大工作量的,需要时间去加,每个架构OP名字还都不一样,可想而知里头需要大量人力去做这些事的。官方会保持rknn toolkit不断的更新来支持更多的op,所以敬请放心。
回复

使用道具 举报

程子

中级会员

积分
386
7#
发表于 2019-2-27 13:37:57 | 只看该作者
RKNN Toolkit 1.0 何时发布?有没有进度规划?
OP数量增长归增长,问题是tf-1.0的OP你们也没都支持啊。
现在tensorflow这边很多自己设计的网络都因OP不支持而无法在RKNN上成功转换,这种感觉像是买了个半成品。
回复

使用道具 举报

raul

中级会员

积分
235
8#
发表于 2019-2-27 14:34:22 | 只看该作者
qiaoqiang126 发表于 2019-2-27 10:19
关于RNN很奇怪,tensorflow里面把Tensor ,unstack小于6个的时候,RNN是可以转成功的。但是UNstack大于6个 ...

没有看到附件中的文件,麻烦再传下,谢谢。
回复

使用道具 举报

qiaoqiang126

中级会员

积分
225
9#
 楼主| 发表于 2019-2-27 17:22:11 | 只看该作者
raul 发表于 2019-2-27 14:34
没有看到附件中的文件,麻烦再传下,谢谢。

见附件

本帖子中包含更多资源

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

x
回复

使用道具 举报

raul

中级会员

积分
235
10#
发表于 2019-2-28 16:51:30 | 只看该作者

我们对unstack的输出个数确实有一些限制,从这个日志来看,unstack好像不止6个。我们现在的限制应该是16个。
回复

使用道具 举报

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

本版积分规则

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


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