facebookresearch/ParlAI

Model export to ONNX

Ouverte

#2 815 ouverte le 6 juil. 2020

Voir sur GitHubGuide contributeur
 (3 commentaires) (1 réaction) (0 personne assignée)Python (2 089 forks)batch import
Help WantedMediumnever-stale

Métriques du dépôt

Stars
 (10 371 étoiles)
Métriques de merge PR
 (Aucune PR mergée en 30 j)

Description

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!

Guide contributeur

Direction de recherche
Commencez par localiser les fichiers de definition de modele dans ParlAI (par exemple, dans parlai/agents/ ou parlai/core/). Recherchez les classes de modele PyTorch qui definissent les modeles de dialogue. Ensuite, suivez le tutoriel d'exportation ONNX de PyTorch pour ajouter une methode d'exportation. Concentrez vous sur la methode forward et les tenseurs d'entree/sortie.
Stack technique
pythonpytorch
Domaine
aibackend
Type d'issue
Documentation
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
3
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-2 jours
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
PythonPyTorch
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
60