From b9ae3e1cfa8a50e2ba09ed42e4bbebfdd0ff96f5 Mon Sep 17 00:00:00 2001 From: lorenzo Date: Fri, 26 Jul 2019 10:32:57 +0200 Subject: [PATCH] small fix --- monoloco/train/trainer.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/monoloco/train/trainer.py b/monoloco/train/trainer.py index 441cedc..a6f6641 100644 --- a/monoloco/train/trainer.py +++ b/monoloco/train/trainer.py @@ -263,13 +263,12 @@ class Trainer: dic_err[phase][clst]['bi'], dic_err[phase][clst]['conf_bi'] * 100)) # Save the model and the results - self.save = False if self.save and not load: torch.save(self.model.state_dict(), self.path_model) print('-'*120) - self.logger.info("model saved: {} \n".format(self.path_model)) + self.logger.info("\nmodel saved: {} \n".format(self.path_model)) else: - self.logger.info("model not saved\n") + self.logger.info("\nmodel not saved\n") return dic_err, self.model