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