Skip to content
Snippets Groups Projects
Commit 0ac54b56 authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[engine.trainer] Flush CSV file as we log it (every epoch); Only output...

[engine.trainer] Flush CSV file as we log it (every epoch); Only output minimal information on the screen
parent 49e1242d
No related branches found
No related tags found
1 merge request!12Streamlining
Pipeline #39231 passed
......@@ -183,7 +183,8 @@ def run(
logdata += gpu_info()
logwriter.writerow(dict(k for k in logdata))
tqdm.write("|".join([f"{k}: {v}" for (k, v) in logdata]))
logfile.flush()
tqdm.write("|".join([f"{k}: {v}" for (k, v) in logdata[:4]]))
total_training_time = time.time() - start_training_time
logger.info(
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment