From 0ac54b56a6a2bcc873cdea7c59ba431fd1480f40 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.anjos@idiap.ch>
Date: Sun, 26 Apr 2020 10:54:58 +0200
Subject: [PATCH] [engine.trainer] Flush CSV file as we log it (every epoch);
Only output minimal information on the screen
---
bob/ip/binseg/engine/trainer.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/bob/ip/binseg/engine/trainer.py b/bob/ip/binseg/engine/trainer.py
index 689b3cca..47f45253 100644
--- a/bob/ip/binseg/engine/trainer.py
+++ b/bob/ip/binseg/engine/trainer.py
@@ -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(
--
GitLab