From 90ccee7fe37793c5c20f33ad45d38984ecddaf5a Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Mon, 31 Aug 2020 11:30:09 +0200
Subject: [PATCH] Fixed a bug: correct number of trained epochs is now
displayed.
---
pysegcnn/core/graphics.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/pysegcnn/core/graphics.py b/pysegcnn/core/graphics.py
index 7c9755e..f8083f2 100644
--- a/pysegcnn/core/graphics.py
+++ b/pysegcnn/core/graphics.py
@@ -375,7 +375,7 @@ def plot_loss(state_file, figsize=(10, 10), step=5,
ls='--', color='grey')
ax1.text(esepoch - nbatches, ax1.get_ylim()[0] + 0.01,
'epoch = {}, accuracy = {:.1f}%'
- .format(int(esepoch / nbatches), esacc * 100),
+ .format(int(esepoch / nbatches) + 1, esacc * 100),
ha='right', color='grey')
# create a patch (proxy artist) for every color
--
GitLab