mrq/vall-e
5
9
Fork 0
Code Issues 8 Pull Requests Packages Projects Releases Wiki Activity

I swear I keep seeing tqdm flicker back a number

Browse Source
This commit is contained in:
mrq 2024-05-10 18:36:01 -05:00
parent 2109712e5b
commit bd0a36ba8d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vall_e/utils/trainer.py
View File

@ -102,7 +102,7 @@ def _non_blocking_input():
def _make_infinite_epochs(dl):
while True:
#_logger.info("New epoch starts.")
yield from tqdm(dl, "Epoch progress", dynamic_ncols=True)
yield from tqdm(dl, "Epoch progress", dynamic_ncols=True, disable=not is_global_leader())
@local_leader_only(default=None)