Fix progress bar

This commit is contained in:
Matthew Honnibal 2017-05-17 14:59:37 +02:00
parent 55dab77de8
commit edfea3a513
1 changed files with 1 additions and 1 deletions

View File

@ -47,7 +47,7 @@ class Trainer(object):
golds = self.make_golds(docs, paragraph_tuples) golds = self.make_golds(docs, paragraph_tuples)
all_docs.extend(docs) all_docs.extend(docs)
all_golds.extend(golds) all_golds.extend(golds)
for batch in tqdm.tqdm(partition_all(12, zip(all_docs, all_golds))): for batch in partition_all(12, zip(tqdm.tqdm(all_docs), all_golds)):
X, y = zip(*batch) X, y = zip(*batch)
yield X, y yield X, y