Toybrick

toolkit load tensorflow的pb模型失败

YEY

注册会员

积分
56
楼主
发表于 2019-12-17 16:11:49    查看: 8953|回复: 4 | [复制链接]    打印 | 只看该作者
拜托大家帮忙看看,我在转一个pb模型,load_tensorflow的时候就出错了,这意思是里面有rknn不支持的层吗?

本帖子中包含更多资源

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

x
回复

使用道具 举报

jefferyzhang

版主

积分
13580
沙发
发表于 2019-12-18 08:36:32 | 只看该作者
从你的log看就是truediv这个op不支持。你可以试着改下模型
回复

使用道具 举报

YEY

注册会员

积分
56
板凳
 楼主| 发表于 2019-12-18 09:34:39 | 只看该作者
jefferyzhang 发表于 2019-12-18 08:36
从你的log看就是truediv这个op不支持。你可以试着改下模型

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

使用道具 举报

jefferyzhang

版主

积分
13580
地板
发表于 2019-12-18 10:03:05 | 只看该作者
YEY 发表于 2019-12-18 09:34
请问这个情况,可以通过自定义op的方式解决吗?哪个版本的toolkit支持呢?

可以自定义op解决。

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

使用道具 举报

YEY

注册会员

积分
56
5#
 楼主| 发表于 2019-12-18 10:06:32 | 只看该作者
jefferyzhang 发表于 2019-12-18 10:03
可以自定义op解决。

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

好的,我试试,非常感谢!
回复

使用道具 举报

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

本版积分规则

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


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