From 5f2089258cf108acf34f397208272b7e291c424e Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.anjos@idiap.ch>
Date: Sat, 4 Apr 2020 21:59:08 +0200
Subject: [PATCH] [engine.trainer] Fix creation of loss evolution
---
bob/ip/binseg/engine/trainer.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/bob/ip/binseg/engine/trainer.py b/bob/ip/binseg/engine/trainer.py
index 8a4b2e50..3201df17 100644
--- a/bob/ip/binseg/engine/trainer.py
+++ b/bob/ip/binseg/engine/trainer.py
@@ -158,8 +158,8 @@ def do_train(
logger.info(f"Total training time: {datetime.timedelta(seconds=total_training_time)} ({(total_training_time/max_epoch):.4f}s in average per epoch)")
# plots a version of the CSV trainlog into a PDF
- logplot_name = os.path.join(output_folder, "trainlog.pdf")
- logdf = pd.read_csv(logplot_name, header=0, names=logfile_fields)
+ logdf = pd.read_csv(logfile_name, header=0, names=logfile_fields)
fig = loss_curve(logdf, title="Loss Evolution")
- logger.info(f"Saving {log_plot_file}")
- fig.savefig(logplot_name)
+ figurefile_name = os.path.join(output_folder, "trainlog.pdf")
+ logger.info(f"Saving {figurefile_name}")
+ fig.savefig(figurefile_name)
--
GitLab