ONNX support for Encoders
cjermain opened this issue · 1 comments
cjermain commented
Based on some initial experiments, it looks like the Encoder classes can not be exported to the ONNX format using torch.onnx.export
. It would be great to be able to package them into an ONNX graph along with the rest of the model. Has anyone looked at this?
I think the relevant operators are available in the ONNX spec, so I may give it a try in the near future.
PetrochukM commented
That'd be cool :) Great idea!