From 463be8e913406b770629252189834a5996425d1f Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Thu, 4 Feb 2021 17:28:37 +0100
Subject: [PATCH] Added logging info when plotting model training statistics.
---
pysegcnn/core/graphics.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/pysegcnn/core/graphics.py b/pysegcnn/core/graphics.py
index 199946c..82badfc 100644
--- a/pysegcnn/core/graphics.py
+++ b/pysegcnn/core/graphics.py
@@ -479,7 +479,9 @@ def plot_loss(state_file, figsize=(10, 10), step=5,
"""
# load the model state
+ state_file = pathlib.Path(state_file)
model_state = torch.load(state_file)
+ LOGGER.info('Plot model statistics: {}'.format(state_file.stem))
# get all non-zero elements, i.e. get number of epochs trained before
# early stop
--
GitLab