facebookresearch / ParlAI

A framework for training and evaluating AI models on a variety of openly available dialogue datasets.
https://parl.ai
MIT License
10.48k stars 2.1k forks source link

Model export to ONNX #2815

Open bakszero opened 4 years ago

bakszero commented 4 years ago

How would one export the trained dialogue models to ONNX? If you can point me to the right files where I can attempt to do something of the sort of what's mentioned in https://pytorch.org/tutorials/advanced/super_resolution_with_onnxruntime.html, it'd be really awesome! Thanks!

stephenroller commented 4 years ago

Honestly, no one on our team has looked into it. If you spent a little time digging and shared feedback or comments, we would welcome them.

github-actions[bot] commented 4 years ago

This issue has not had activity in 30 days. Please feel free to reopen if you have more issues. You may apply the "never-stale" tag to prevent this from happening.

Darth-Carrotpie commented 2 years ago

So did anyone dig into it? Is it doable? I see people wanting a speed increase (me including), i.e. #4257. ONNX does that perfectly, plus gives chance reusability in other platforms / languages. Honestly, it feels like it should be a priority task...