facebookresearch/ParlAI

Model export to ONNX

Open

#2.815 geöffnet am 6. Juli 2020

Auf GitHub ansehen
 (3 Kommentare) (1 Reaktion) (0 zugewiesene Personen)Python (2.089 Forks)batch import
Help WantedMediumnever-stale

Repository-Metriken

Stars
 (10.371 Stars)
PR-Merge-Metriken
 (Keine gemergten PRs in 30 T)

Beschreibung

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!

Contributor Guide

Research-Richtung
Beginnen Sie damit, die Modelldefinitionsdateien in ParlAI zu finden (z.B. in parlai/agents/ oder parlai/core/). Suchen Sie nach PyTorch Modellklassen, die die Dialogmodelle definieren. Folgen Sie dann dem PyTorch ONNX Export Tutorial, um eine Exportmethode hinzuzufügen. Konzentrieren Sie sich auf die forward Methode und die Ein /Ausgabe Tensoren.
Tech Stack
pythonpytorch
Domain
aibackend
Issue Type
Dokumentation
SchwierigkeitGeschätzte Implementierungsschwierigkeit für neue Contributors, von 1 für sehr kleine Änderungen bis 5 für Expertenarbeit.
3
Geschätzte ZeitEin grober Zeitrahmen, um zu recherchieren, zu implementieren, zu testen und einen Pull Request vorzubereiten.
1-2 Tage
AktivitätsstatusWie verfügbar das Issue gerade wirkt: frisch, aktiv, stale, blockiert oder wartet auf Maintainer-Input.
Aktiv
KlarheitWie klar das Issue die erwartete Änderung, Akzeptanzkriterien und den nächsten Schritt erklärt.
Klar
Voraussetzungen
PythonPyTorch
EinsteigerfreundlichkeitEin Score von 1 bis 100, der schätzt, wie zugänglich dieses Issue für First-Time Contributors ist.
60