Toybrick

pb转rknn量化时,NoneType问题

kiwigogogo

注册会员

积分
86
发表于 2019-9-10 20:21:41    查看: 7545|回复: 4 | [复制链接]    打印 | 显示全部楼层
自己training一个pb模型,进行量化时do_quantization=True,结果报错了。是因为rknn-toolkit不支持truediv、Realdiv运算吗?我用inception pb model量化转rknn model是可以的。


本帖子中包含更多资源

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

x
回复

使用道具 举报

jefferyzhang

版主

积分
13083
发表于 2019-9-11 08:15:55 | 显示全部楼层
从log上看是的
回复

使用道具 举报

kiwigogogo

注册会员

积分
86
 楼主| 发表于 2019-9-11 09:03:34 | 显示全部楼层

类似的不支持某个层的报错,应该怎么解决呢?修改我自己的tf层,还是等官方发布新的toolkit支持?
回复

使用道具 举报

jefferyzhang

版主

积分
13083
发表于 2019-9-11 09:06:41 | 显示全部楼层
kiwigogogo 发表于 2019-9-11 09:03
类似的不支持某个层的报错,应该怎么解决呢?修改我自己的tf层,还是等官方发布新的toolkit支持? ...

1.2的 toolkit已经支持自定义op,你可以按着文档方法自己适配
当然也可以修改自己模型绕过这些op
回复

使用道具 举报

kiwigogogo

注册会员

积分
86
 楼主| 发表于 2019-9-11 10:49:58 | 显示全部楼层
本帖最后由 kiwigogogo 于 2019-9-11 17:35 编辑
jefferyzhang 发表于 2019-9-11 09:06
1.2的 toolkit已经支持自定义op,你可以按着文档方法自己适配
当然也可以修改自己模型绕过这些op ...

谢谢,我目前用的是v1.1.0,我下载rknn-toolkit-v1.2来试试。
回复

使用道具 举报

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

本版积分规则

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


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