Can the HTS-AT model be exported as ONNX？

[jonathan3chen]

Can the HTS-AT model be exported as ONNX？I have some problem when exporting the sed_model to onnx using torch.onnx.export

[RetroCirce]

Hi,

I think this is doable, I converted to the torchscript once upon the time. I believe you might need to update the library to the relative new one (like pytorch 2.0, 2.2). And to see if you can use the pytorch utility functions to do it.

[VladSkripniuk]

Hi,

Thank you for the great work on HTS-AT!

Would it be possible for you to release a pre-exported ONNX checkpoint of the model? It would greatly simplify integration for use cases like mine, and I’m sure others in the community would find it useful as well.

If you have any recommendations or a working export script, I’d also appreciate that.

Thanks again for your contributions!

[rabeya-akter]

Hello,
Can you share the ONNX checkpoint or code to generate it?
Thank you.

[RetroCirce]

Hi all,

Thank you for all of your feedback on this repo.
Due to my work schedule and policy restriction, I am not able to provide the onnx model. But I would release a pip library of HTS-AT and upgrade HTS-AT to support the latest pytorch version (from 1.13 --> 2.7). The estimate release date will be May 15th.