nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-11 05:13:31 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

[fix] extract feat log

Browse source
This commit is contained in:
zhengzangw 2024-05-17 10:05:49 +00:00
parent 5aad90acaf
commit 5a1cf0c718
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
scripts/misc/extract_feat.py
View file

@ -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)