SHI-Labs/Neighborhood-Attention-Transformer

ONNX

idanpd opened this issue · 2 comments

idanpd commented

Hi,
trying to convert to onnx but natten module can not be converted.
Is there any solution for this kind of problem ?

issue created when getting to:

_C.natten2dav_forward(
attn,
value,
dilation)

alihassanijr commented

Hello and thank you for your interest.

Unfortunately that is not surprising, because Neighborhood Attention is a "custom" torch operation, and therefore doesn't support conversion to ONNX.
In theory we --could-- try and avoid the error, but then the problem would be that you'd still need to run that ONNX model on a PyTorch environment only, which kind of defeats the purpose.

I would however recommend raising this issue in NATTEN, because that's where we've moved our entire extension, and therefore this repository only uses that package. As long as NATTEN starts supporting ONNX, the issue would be resolved for models found in this repository. However, even if it does, it'll still have to be implemented for your target framework in order to run there.

I hope this answers your question.

alihassanijr commented

Closing this issue since it exists in NATTEN: SHI-Labs/NATTEN#17 .