diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py index 271f77f05a0686b6124d70bd321aa15a6137a888..4fc4f2f0d5ab65267e297e40d7f53b379ac057f6 100644 --- a/pysegcnn/core/trainer.py +++ b/pysegcnn/core/trainer.py @@ -2674,9 +2674,12 @@ class NetworkInference(BaseConfig): # check whether to aggregate the results of the different model runs if self.aggregate: + LOGGER.info('Aggregating statistics of models:') + LOGGER.info(('\n ' + (len(__name__) + 1) * ' ').join( + ['{}'.format(mstate.name) for mstate in self.state_files])) # base name for all models - base_name = str(self.state_files[0]).name + base_name = str(self.state_files[0].name) # chech whether to compute the aggregated confusion matrix if self.cm: