1
0
Fork 0

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

master
mrq 2023-02-19 06:28:53 +07:00
parent debdf6049a
commit 485319c2bb
1 changed files with 2 additions and 1 deletions

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