pytorch/examples

storing the best model

Open

#232 aperta il 12 ott 2017

Vedi su GitHub
 (5 commenti) (0 reazioni) (0 assegnatari)Python (9429 fork)batch import
good first issue

Metriche repository

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

Descrizione

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

Guida contributor

Direzione di ricerca
Il problema si riferisce alla riga 171 in word language model/main.py (https://github.com/pytorch/examples/blob/7d0d413425e2ee64fcd0e0de1b11c5cca1f79f4d/word language model/main.py#L171). Il codice attuale salva il modello alla fine di ogni epoca, ma il problema afferma che salva il modello corrente, non quello migliore. Per risolvere, aggiungere una logica per tracciare la miglior perdita di validazione e salvare il modello solo quando viene raggiunto un nuovo miglior risultato. Assicurarsi che lo stato del modello venga salvato correttamente utilizzando torch.save. La modifica dovrebbe essere localizzata al ciclo di addestramento attorno a quella riga. Nessuna PR collegata o commenti recenti indicano lavoro attivo.
Tech stack
pythonpytorch
Dominio
machine learningai
Tipo issue
Bug
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
GitPythonPyTorch
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
75