facebookresearch/ParlAI

Model export to ONNX

Open

#2,815 建立於 2020年7月6日

在 GitHub 查看
 (3 留言) (1 反應) (0 負責人)Python (2,089 fork)batch import
Help WantedMediumnever-stale

倉庫指標

Star
 (10,371 star)
PR 合併指標
 (30 天內沒有已合併 PR)

描述

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!

貢獻者指南

研究方向
首先在 ParlAI 中定位模型定義檔案（例如 parlai/agents/ 或 parlai/core/）。找到定義對話模型的 PyTorch 模型類別。然後按照 PyTorch ONNX 匯出教學新增匯出方法。重點關注 forward 方法和輸入/輸出張量。
技術棧
pythonpytorch
領域
aibackend
議題類型
文件
難度面向新貢獻者的預計實作難度，1 表示很小改動，5 表示專家級工作。
3
預計時間有經驗貢獻者完成調查、實作、測試並準備 pull request 的粗略時間範圍。
1-2 天
活動狀態議題目前的可參與程度：新鮮、活躍、陳舊、阻塞或等待維護者輸入。
活躍
清晰度議題是否清楚說明預期改動、驗收標準和下一步。
清晰
前置要求
PythonPyTorch
新手友善度1-100 的估計分數，表示該議題對首次貢獻者的友善程度。
60