标题: pytorch aten hardwish operator support [打印本页] 作者: rakidis 时间: 2021-9-6 19:11 标题: pytorch aten hardwish operator support 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.
作者: jefferyzhang 时间: 2021-9-6 21:59
You can try using a custom OP to support it.
Plz see the documentation for more information.作者: rakidis 时间: 2021-9-6 22:13
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