facebookresearch/ParlAI

Model export to ONNX

Aberta

#2.815 aberto em 6 de jul. de 2020

Ver no GitHubGuia do colaborador
 (3 comentários) (1 reação) (0 responsável)Python (2.089 forks)batch import
Help WantedMediumnever-stale

Métricas do repositório

Stars
 (10.371 estrelas)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 30d)

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!

Guia do colaborador

Direção de pesquisa
Comece localizando os arquivos de definição de modelo no ParlAI (por exemplo, em parlai/agents/ ou parlai/core/). Procure pelas classes de modelo PyTorch que definem os modelos de diálogo. Em seguida, siga o tutorial de exportação ONNX do PyTorch para adicionar um método de exportação. Concentre se no método forward e nos tensores de entrada/saída.
Pilha de tecnologia
pythonpytorch
Domain
aibackend
Tipo Issue
Documentação
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
3
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-2 dias
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
PythonPyTorch
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
60