Skip to content
This repository has been archived by the owner on Jul 4, 2023. It is now read-only.

ONNX support for Encoders #95

Open
cjermain opened this issue Mar 4, 2020 · 1 comment
Open

ONNX support for Encoders #95

cjermain opened this issue Mar 4, 2020 · 1 comment

Comments

@cjermain
Copy link

cjermain commented Mar 4, 2020

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
Copy link
Owner

That'd be cool :) Great idea!

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants