diff --git a/scripts/question_answering/run_squad.py b/scripts/question_answering/run_squad.py
index 521ee15a47..b0b6da70e4 100644
--- a/scripts/question_answering/run_squad.py
+++ b/scripts/question_answering/run_squad.py
@@ -864,7 +864,7 @@ def eval_validation(ckpt_name, best_eval):
         all_results = []
         epoch_tic = time.time()
         tic = time.time()
-        epoch_size = len(dev_features)
+        epoch_size = len(dev_all_chunk_features)
         total_num = 0
         log_num = 0
         for batch_idx, dev_batch in enumerate(grouper(dev_dataloader, len(ctx_l))):