diff --git a/ppcls/engine/evaluation/retrieval.py b/ppcls/engine/evaluation/retrieval.py
index 1b99237f4c23806c58787291916b69c9856c30ec..4d823809762aaad741497bc7ff3ed486084ee631 100644
--- a/ppcls/engine/evaluation/retrieval.py
+++ b/ppcls/engine/evaluation/retrieval.py
@@ -125,12 +125,8 @@ def compute_feature(engine, name="gallery"):
     all_feat = []
     all_label = []
     all_camera = []
-    max_iter = len(dataloader) - 1 if platform.system() == "Windows" else len(
-        dataloader)
     has_camera = False
     for idx, batch in enumerate(dataloader):  # load is very time-consuming
-        if idx >= max_iter:
-            break
         if idx % engine.config["Global"]["print_batch_step"] == 0:
             logger.info(
                 f"{name} feature calculation process: [{idx}/{len(dataloader)}]"