facebookresearch/fairseq

Request to provide torchscript-ed model

Aberta

#4.853 aberto em 9 de nov. de 2022

Ver no GitHubGuia do colaborador
 (0 comentário) (1 reação) (0 responsável)Python (6.224 forks)batch import
enhancementhelp wantedneeds triage

Métricas do repositório

Stars
 (29.107 estrelas)
Métricas de merge de PR
 (Métricas PR pendentes)

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.

Guia do colaborador

Direção de pesquisa
Explore o código fonte do fairseq para entender como instanciar o modelo wav2vec2 e carregar o checkpoint pré treinado. Estude a documentação do PyTorch TorchScript para aprender como scriptar ou rastrear o modelo. Tente exportar o modelo e verifique se a saída corresponde ao original. Por fim, revise o processo de upload do modelo no Hugging Face.
Pilha de tecnologia
pythonpytorch
Domain
machine learningai
Tipo Issue
Funcionalidade
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
2
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-3 horas
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
PythonPyTorchGit
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
80