pytorch/examples

storing the best model

Open

#232 aberto em 12 de out. de 2017

Ver no GitHub
 (5 comments) (0 reactions) (0 assignees)Python (9.429 forks)batch import
good first issue

Métricas do repositório

Stars
 (21.634 stars)
Métricas de merge de PR
 (Nenhuma PRs mesclada em 30d)

Description

https://github.com/pytorch/examples/blob/7d0d413425e2ee64fcd0e0de1b11c5cca1f79f4d/word_language_model/main.py#L171 doesn't look like it is actually saving the best model? It saves only the current model

Guia do colaborador

Direção de pesquisa
O problema aponta para a linha 171 em word language model/main.py (https://github.com/pytorch/examples/blob/7d0d413425e2ee64fcd0e0de1b11c5cca1f79f4d/word language model/main.py#L171). O código atual salva o modelo no final de cada época, mas o problema afirma que ele salva o modelo atual, não o melhor. Para corrigir, adicione lógica para rastrear a melhor perda de validação e salve o modelo apenas quando um novo melhor for alcançado. Certifique se de que o estado do modelo seja salvo corretamente usando torch.save. A alteração deve ser localizada no loop de treinamento em torno dessa linha. Nenhum PR vinculado ou comentários recentes indicam trabalho ativo.
Pilha de tecnologia
pythonpytorch
Domain
machine learningai
Tipo Issue
Bug
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
GitPythonPyTorch
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
75