Toybrick

标题: pb转rknn量化时，NoneType问题 [打印本页]

作者: kiwigogogo 时间: 2019-9-10 20:21
标题: pb转rknn量化时，NoneType问题
自己training一个pb模型，进行量化时do_quantization=True，结果报错了。是因为rknn-toolkit不支持truediv、Realdiv运算吗？我用inception pb model量化转rknn model是可以的。

[attach]553[/attach]

作者: jefferyzhang 时间: 2019-9-11 08:15
从log上看是的

作者: kiwigogogo 时间: 2019-9-11 09:03

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

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

作者: jefferyzhang 时间: 2019-9-11 09:06

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

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

作者: kiwigogogo 时间: 2019-9-11 10:49
本帖最后由 kiwigogogo 于 2019-9-11 17:35 编辑

jefferyzhang 发表于 2019-9-11 09:06
1.2的 toolkit已经支持自定义op，你可以按着文档方法自己适配
当然也可以修改自己模型绕过这些op ...

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

欢迎光临 Toybrick (https://t.rock-chips.com/)