From 5a1cf0c718a25f04490c0180cbfff12e5c0b1570 Mon Sep 17 00:00:00 2001 From: zhengzangw Date: Fri, 17 May 2024 10:05:49 +0000 Subject: [PATCH] [fix] extract feat log --- scripts/misc/extract_feat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/misc/extract_feat.py b/scripts/misc/extract_feat.py index ac2bef9..0964414 100644 --- a/scripts/misc/extract_feat.py +++ b/scripts/misc/extract_feat.py @@ -133,7 +133,7 @@ def main(): # == training loop in an epoch == dataloader_iter = iter(dataloader) log_time = cfg.get("log_time", False) - for i in tqdm(range(0, num_bin_to_process)): + for i in tqdm(range(0, num_bin_to_process * bin_size)): with Timer("step", log=log_time): with Timer("data loading", log=log_time): batch = next(dataloader_iter)