From 77944d1b313ad532d7f60cd8e7355af04617e4b5 Mon Sep 17 00:00:00 2001 From: Yuqiang Xie Date: Thu, 21 Mar 2019 20:59:33 +0800 Subject: [PATCH] add tqdm to the process of eval Maybe better. --- examples/run_swag.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/run_swag.py b/examples/run_swag.py index ab8e147cae..f193582640 100644 --- a/examples/run_swag.py +++ b/examples/run_swag.py @@ -509,7 +509,7 @@ def main(): model.eval() eval_loss, eval_accuracy = 0, 0 nb_eval_steps, nb_eval_examples = 0, 0 - for input_ids, input_mask, segment_ids, label_ids in eval_dataloader: + for input_ids, input_mask, segment_ids, label_ids in tqdm(eval_dataloader, desc="Evaluating"): input_ids = input_ids.to(device) input_mask = input_mask.to(device) segment_ids = segment_ids.to(device)