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

Fix pretrain

This commit is contained in:
Matthew Honnibal 2018-11-15 22:46:53 +00:00
parent 6af6950e46
commit f8afaa0c1c
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
spacy/cli/pretrain.py
View File

@ -188,8 +188,8 @@ def pretrain(texts_loc, vectors_model, output_dir, width=128, depth=4,
progress = tracker.update(epoch, loss, docs)
if progress:
print(*progress)
if texts_loc == '-' and progress.words_per_epoch[epoch] >= 10**7:
break
if texts_loc == '-' and tracker.words_per_epoch[epoch] >= 10**7:
break
with model.use_params(optimizer.averages):
with (output_dir / ('model%d.bin' % epoch)).open('wb') as file_:
file_.write(tok2vec.to_bytes())