Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix averages

This commit is contained in:
Matthew Honnibal 2020-09-02 19:37:43 +02:00
parent 737a1408d9
commit 122cb02001
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/cli/train.py
View File

@ -159,7 +159,8 @@ def train(
print_row(info) print_row(info)
if is_best_checkpoint and output_path is not None: if is_best_checkpoint and output_path is not None:
update_meta(T_cfg, nlp, info) update_meta(T_cfg, nlp, info)
nlp.to_disk(output_path / "model-best") with nlp.use_params(optimizer.averages):
nlp.to_disk(output_path / "model-best")
progress = tqdm.tqdm(total=T_cfg["eval_frequency"], leave=False) progress = tqdm.tqdm(total=T_cfg["eval_frequency"], leave=False)
progress.set_description(f"Epoch {info['epoch']}") progress.set_description(f"Epoch {info['epoch']}")
except Exception as e: except Exception as e: