From 9c7eb6877caaecffc6da1d9fbfe0013cd40aac87 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Fri, 5 Feb 2021 15:40:05 +0100
Subject: [PATCH] Improved logging for inference.
---
pysegcnn/core/trainer.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index f3089ff..8943cec 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2617,8 +2617,6 @@ class NetworkInference(BaseConfig):
# load existing model evaluation
if not self.overwrite:
- LOGGER.info('Using model evaluation: {}.'
- .format(self.eval_file(state)))
inference[state.stem] = torch.load(self.eval_file(state))
continue
else:
@@ -2666,6 +2664,8 @@ class NetworkInference(BaseConfig):
outpath=self.perfmc_path)
# save model predictions to file
+ LOGGER.info('Saving model evaluation: {}'
+ .format(self.eval_file(state)))
torch.save(output, self.eval_file(state))
# save model predictions to list
--
GitLab