diff --git a/ppcls/engine/evaluation/retrieval.py b/ppcls/engine/evaluation/retrieval.py index ef4bbd24c..54ef8e46f 100644 --- a/ppcls/engine/evaluation/retrieval.py +++ b/ppcls/engine/evaluation/retrieval.py @@ -146,11 +146,7 @@ def cal_feature(engine, name='gallery'): batch_feas_list = [] img_id_list = [] unique_id_list = [] - max_iter = len(dataloader) - 1 if platform.system() == "Windows" else len( - dataloader) 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)}]"