charles/monoloco
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

small fix

Browse Source
This commit is contained in:
lorenzo 2019-07-26 10:32:57 +02:00
parent 87ec4c441e
commit b9ae3e1cfa
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
monoloco/train/trainer.py
View File

@ -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