toolkit load tensorflow的pb模型失败

[复制链接] · 发表于 2019-12-17 16:11:49

拜托大家帮忙看看，我在转一个pb模型，load_tensorflow的时候就出错了，这意思是里面有rknn不支持的层吗？

只看该作者 · 发表于 2019-12-18 08:36:32

从你的log看就是truediv这个op不支持。你可以试着改下模型

只看该作者 · 发表于 2019-12-18 09:34:39

jefferyzhang 发表于 2019-12-18 08:36
从你的log看就是truediv这个op不支持。你可以试着改下模型

请问这个情况，可以通过自定义op的方式解决吗？哪个版本的toolkit支持呢?

只看该作者 · 发表于 2019-12-18 10:03:05

YEY 发表于 2019-12-18 09:34
请问这个情况，可以通过自定义op的方式解决吗？哪个版本的toolkit支持呢?

可以自定义op解决。

toolkit我们都是建议用最新，目前最新就是1.2.1，新的都会解决前一个版本bug和支持新功能

只看该作者 · 发表于 2019-12-18 10:06:32

jefferyzhang 发表于 2019-12-18 10:03
可以自定义op解决。

toolkit我们都是建议用最新，目前最新就是1.2.1，新的都会解决前一个版本bug和支持 ...

好的，我试试，非常感谢！