From bd0a36ba8ddafd14d71c8847c52ecd3cc0bd337f Mon Sep 17 00:00:00 2001 From: mrq Date: Fri, 10 May 2024 18:36:01 -0500 Subject: [PATCH] I swear I keep seeing tqdm flicker back a number --- vall_e/utils/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vall_e/utils/trainer.py b/vall_e/utils/trainer.py index 21e061b..7d4b3b5 100755 --- a/vall_e/utils/trainer.py +++ b/vall_e/utils/trainer.py @@ -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)