Toybrick

标题: toolkit load tensorflow的pb模型失败 [打印本页]

作者: YEY 时间: 2019-12-17 16:11
标题: toolkit load tensorflow的pb模型失败
拜托大家帮忙看看，我在转一个pb模型，load_tensorflow的时候就出错了，这意思是里面有rknn不支持的层吗？

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

作者: YEY 时间: 2019-12-18 09:34

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

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

作者: jefferyzhang 时间: 2019-12-18 10:03

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

可以自定义op解决。

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

作者: YEY 时间: 2019-12-18 10:06

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

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

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

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