pb转rknn量化时，NoneType问题

[复制链接] · 发表于 2019-9-10 20:21:41

自己training一个pb模型，进行量化时do_quantization=True，结果报错了。是因为rknn-toolkit不支持truediv、Realdiv运算吗？我用inception pb model量化转rknn model是可以的。

只看该作者 · 发表于 2019-9-11 08:15:55

从log上看是的

只看该作者 · 发表于 2019-9-11 09:03:34

jefferyzhang 发表于 2019-9-11 08:15
从log上看是的

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

只看该作者 · 发表于 2019-9-11 09:06:41

kiwigogogo 发表于 2019-9-11 09:03
类似的不支持某个层的报错，应该怎么解决呢？修改我自己的tf层，还是等官方发布新的toolkit支持？ ...

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

只看该作者 · 发表于 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来试试。