c-my/transformers
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add tqdm, clean up logging

Browse Source
This commit is contained in:
thomwolf committed 2018-11-04 11:07:34 +01:00
1 parent d4e3cf3520
commit 0b7a20c651
2 files changed
+2 -4
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

run_squad.py
+2 -2
View File
@@ -912,9 +912,9 @@ def main():
model.eval()
all_results = []
logger.info("Start evaulating")
logger.info("Start evaluating")
#for input_ids, input_mask, segment_ids, label_ids, example_index in eval_dataloader:
for input_ids, input_mask, segment_ids, example_index in eval_dataloader:
for input_ids, input_mask, segment_ids, example_index in tqdm(eval_dataloader, descr="Evaluating"):
if len(all_results) % 1000 == 0:
logger.info("Processing example: %d" % (len(all_results)))