Skip to content

GitLab

Unverified Commit 77944d1b authored by Yuqiang Xie's avatar Yuqiang Xie Committed by GitHub
Browse files

add tqdm to the process of eval 

Maybe better.
parent f3e54048
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/run_swag.py
View file @ 77944d1b
...@@ -509,7 +509,7 @@ def main(): ...@@ -509,7 +509,7 @@ def main():
model.eval() model.eval()
eval_loss, eval_accuracy = 0, 0 eval_loss, eval_accuracy = 0, 0
nb_eval_steps, nb_eval_examples = 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_ids = input_ids.to(device)
input_mask = input_mask.to(device) input_mask = input_mask.to(device)
segment_ids = segment_ids.to(device) segment_ids = segment_ids.to(device)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment