diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py index f3089ff04f3d6ef9d1c9714e8f425a077fc9e96a..8943cecd7009608b10ced44146ad140b2ffcb2c2 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