facebookresearch/fairseq

Request to provide torchscript-ed model

Open

#4 853 ouverte le 9 nov. 2022

Voir sur GitHub
 (0 commentaires) (1 réaction) (0 assignés)Python (6 224 forks)batch import
enhancementhelp wantedneeds triage

Métriques du dépôt

Stars
 (29 107 stars)
Métriques de merge PR
 (Aucune PR mergée en 30 j)

Description

🚀 Feature Request

https://huggingface.co/facebook/wav2vec2-base-960h/tree/main provides only model.state_dict(). Please also consider upload a torchscript-ed model using either torch.jit.script() or torch.jit.trace().

Motivation

It simplifies users' life as they don't need to install fairseq to use a torchscript-ed model

Additional context

We want to test models trained using CTC loss from various frameworks using k2 for decoding. (See https://github.com/k2-fsa/k2/pull/1096)

We only need a torchscript-ed model but for the current situation, we have to install fairseq and export the pretrained model by ourselves via torchscript.

Guide contributeur

Direction de recherche
Explorez la base de code de fairseq pour comprendre comment instancier le modèle wav2vec2 et charger le point de contrôle pré entraîné. Étudiez la documentation PyTorch TorchScript pour apprendre à scripter ou tracer le modèle. Essayez d'exporter le modèle et vérifiez que sa sortie correspond à l'original. Enfin, examinez le processus de téléchargement du modèle Hugging Face.
Stack technique
pythonpytorch
Domaine
machine learningai
Type d'issue
Fonctionnalité
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
2
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-3 heures
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
PythonPyTorchGit
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
80