facebookresearch/ParlAI

Model export to ONNX

Open

#2815 aperta il 6 lug 2020

Vedi su GitHub
 (3 commenti) (1 reazione) (0 assegnatari)Python (2089 fork)batch import
Help WantedMediumnever-stale

Metriche repository

Star
 (10.371 star)
Metriche merge PR
 (Nessuna PR mergiata in 30 g)

Descrizione

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!

Guida contributor

Direzione di ricerca
Inizia individuando i file di definizione del modello in ParlAI (ad esempio, in parlai/agents/ o parlai/core/). Cerca le classi del modello PyTorch che definiscono i modelli di dialogo. Quindi segui il tutorial di esportazione ONNX di PyTorch per aggiungere un metodo di esportazione. Concentrati sul metodo forward e sui tensori di input/output.
Tech stack
pythonpytorch
Dominio
aibackend
Tipo issue
Documentazione
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-2 giorni
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
PythonPyTorch
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
60