Toybrick

请问 rknn 是否支持 tanh 激活函数?

leok

版主

积分
894
楼主
发表于 2020-6-2 17:11:57 | 显示全部楼层
Aiden 发表于 2020-6-2 16:48
Hi, 我的试验环境是:

rknn-toolkit           1.3.2

1、首先,在pytorch保存部署模型时,需要调eval,去把模型中变量固化。
2、rknn不是单独的某个op或者activation去支持,内部会进行优化组合,提高效率。
回复

使用道具 举报

leok

版主

积分
894
沙发
发表于 2020-6-2 18:12:58 | 显示全部楼层
Aiden 发表于 2020-6-2 17:39
leok, 谢谢回复。

在pytorch-->rknn路走不通的情况下;可以尝试pytorch-->onnx-->rknn。
回复

使用道具 举报

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

本版积分规则

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


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