My team has trained a model whose upstream model was DistilHuBERT. We’re happy with the outcomes, and we’d like to convert it to ONNX in order to deploy to multiple platforms.
We obtain the following error when we try the conversion with
RuntimeError: ONNX export failed: Couldn't export Python operator GradMultiply
Has anyone else run into this issue? What’s the best practice - ask the ONNX team to implement
GradMultiply, or is there already some workarounds we can apply?