From 485319c2bb25e291868f385180e7bce0db4aa6fe Mon Sep 17 00:00:00 2001 From: mrq Date: Sun, 19 Feb 2023 06:28:53 +0000 Subject: [PATCH] don't know what brain worms had me throw printing training output under verbose --- src/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/utils.py b/src/utils.py index 643e82f..5a582f1 100755 --- a/src/utils.py +++ b/src/utils.py @@ -496,8 +496,9 @@ def run_training(config_path, verbose=False, buffer_size=8, progress=gr.Progress checkpoint = checkpoint + 1 progress(checkpoint / float(checkpoints), f'[{checkpoint}/{checkpoints}] Saving checkpoint...') + print(f"[Training] [{datetime.now().isoformat()}] {line[:-1]}") + if verbose: - print(f"[Training] [{datetime.now().isoformat()}] {line[:-1]}") yield "".join(buffer[-buffer_size:]) training_process.stdout.close()