Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix logging error (#575) 

* fix logging error

* no need for the '+' sign

* move space to beginning of next line
This commit is contained in:
Dang Nguyen Anh Khoa 2019-12-04 19:04:14 +07:00 committed by William Falcon
parent ab4fea0b55
commit b5b77e44b1
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pytorch_lightning/callbacks/pt_callbacks.py
View File

@ -312,16 +312,16 @@ class ModelCheckpoint(Callback):
self.best = max(self.best_k_models.values()) self.best = max(self.best_k_models.values())
if self.verbose > 0: if self.verbose > 0:
logging.info( logging.info(
f'\nEpoch {epoch:05d}: {self.monitor} reached', f'\nEpoch {epoch:05d}: {self.monitor} reached'
f'{current:0.5f} (best {self.best:0.5f}), saving model to', f' {current:0.5f} (best {self.best:0.5f}), saving model to'
f'{filepath} as top {self.save_top_k}') f' {filepath} as top {self.save_top_k}')
self._save_model(filepath) self._save_model(filepath)
else: else:
if self.verbose > 0: if self.verbose > 0:
logging.info( logging.info(
f'\nEpoch {epoch:05d}: {self.monitor}', f'\nEpoch {epoch:05d}: {self.monitor}'
f'was not in top {self.save_top_k}') f' was not in top {self.save_top_k}')
else: else:
if self.verbose > 0: if self.verbose > 0: