Toybrick

pytorch aten hardwish operator support

rakidis

高级会员

积分
507
楼主
发表于 2021-9-6 19:11:37    查看: 3062|回复: 2 | [复制链接]    打印 | 只看该作者
Hello,
When I try to convert a pytorch model with 1.7.0 rknn toolkit, I get the error shown below.

NotImplementedError: The following operators are not implemented: ['aten::hardswish_']

Do you have plan to support this hardswish operator in near future? I saw on twm github that they added support for it.

https://github.com/apache/tvm/pull/7174

Do you have plan to merge it rknn toolkit?

Regards


回复

使用道具 举报

jefferyzhang

版主

积分
12964
沙发
发表于 2021-9-6 21:59:39 | 只看该作者
You can try using a custom OP to support it.
Plz see the documentation for more information.
回复

使用道具 举报

rakidis

高级会员

积分
507
板凳
 楼主| 发表于 2021-9-6 22:13:51 | 只看该作者
That's a No I guess then.

Custom Ops are very complicated to implement unfortunately.

Many new pytorch projects use aten operations in their solutions, rockchip must try to support most aten operations
回复

使用道具 举报

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

本版积分规则

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


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