提交 f54f03ac 编写于 作者: H HydrogenSulfate 提交者: Walter

fix(retrieval): remove max_iter in retrieval.py to avoid empty data

上级 0c6749b7
......@@ -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)}]"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册