Use averaged params for evaluation

2017-05-21 17:49:46 -05:00 · 2017-05-21 17:49:46 -05:00 · e14533757b
parent 7811d97339
commit e14533757b
1 changed files with 2 additions and 1 deletions
--- a/spacy/cli/train.py
+++ b/spacy/cli/train.py
@ -53,6 +53,7 @@ def train(lang_id, output_dir, train_data, dev_data, n_iter, n_sents,
                golds = list(golds)
                nlp.update(docs, golds, drop=dropout, sgd=optimizer)
                pbar.update(len(docs))
+        with nlp.use_params(optimizer.averages):
            scorer = nlp.evaluate(corpus.dev_docs(nlp))
        print_progress(i, {}, scorer.scores)
    with (output_path / 'model.bin').open('wb') as file_: