add tqdm to the process of eval

Maybe better.
2019-03-21 20:59:33 +08:00
parent f3e5404880
commit 77944d1b31
1 changed files with 1 additions and 1 deletions
--- 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)