diff --git a/examples/run_swag.py b/examples/run_swag.py index ab8e147c..f1935826 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)