diff --git a/bob/ip/binseg/engine/trainer.py b/bob/ip/binseg/engine/trainer.py index 689b3cca4183dd48790811e595ba369ee197af6d..47f452535e2c05dabe904310690f87b8a38336c3 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(