Move comment

This commit is contained in:
shademe 2022-12-22 17:16:38 +01:00
parent 7e793ded5a
commit 2dc82234f4
No known key found for this signature in database
GPG Key ID: 6FCA9FC635B2A402

View File

@ -171,7 +171,6 @@ def console_logger_v3(
)
)
if progress_bar:
# Set disable=None, so that it disables on non-TTY
if progress_bar == "train":
total = max_steps
desc = f"Last Eval Epoch: {info['epoch']}"
@ -180,6 +179,7 @@ def console_logger_v3(
total = eval_frequency
desc = f"Epoch {info['epoch']+1}"
initial = 0
# Set disable=None, so that it disables on non-TTY
progress = tqdm.tqdm(
total=total,
disable=None,