facebookresearch/fairseq

Request to provide torchscript-ed model

Open

#4853 aperta il 9 nov 2022

Vedi su GitHub
 (0 commenti) (1 reazione) (0 assegnatari)Python (6224 fork)batch import
enhancementhelp wantedneeds triage

Metriche repository

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

Descrizione

🚀 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.

Guida contributor

Direzione di ricerca
Esplora il codebase di fairseq per capire come istanziare il modello wav2vec2 e caricare il checkpoint pre addestrato. Studia la documentazione di PyTorch TorchScript per imparare come scrivere script o tracciare il modello. Prova a esportare il modello e verifica che l'output corrisponda all'originale. Infine, rivedi il processo di upload del modello su Hugging Face.
Tech stack
pythonpytorch
Dominio
machine learningai
Tipo issue
Funzionalità
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
2
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-3 ore
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
PythonPyTorchGit
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
80