Unbreak progress bar

This commit is contained in:
Matthew Honnibal 2018-12-20 13:57:00 +01:00
parent c315e08e6e
commit 1788bf1af7

View File

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