don't know what brain worms had me throw printing training output under verbose

This commit is contained in:
mrq 2023-02-19 06:28:53 +00:00
parent debdf6049a
commit 485319c2bb

View File

@ -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()