Unbreak progress bar

This commit is contained in:
Matthew Honnibal 2018-12-20 13:57:00 +01:00
parent c315e08e6e
commit 1788bf1af7
1 changed files with 1 additions and 1 deletions

View File

@ -221,7 +221,7 @@ def train(
(nlp.make_doc(rt["text"]) for rt in raw_text), size=8 (nlp.make_doc(rt["text"]) for rt in raw_text), size=8
) )
words_seen = 0 words_seen = 0
with _create_progress_bar(n_train_words) as pbar: with tqdm.tqdm(total=n_train_words, leave=False) as pbar:
losses = {} losses = {}
for batch in util.minibatch_by_words(train_docs, size=batch_sizes): for batch in util.minibatch_by_words(train_docs, size=batch_sizes):
if not batch: if not batch: