Can the HTS-AT model be exported as ONNX？

[chenyangzhen]

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.