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

fixed bad warn for result obj (#3072)

This commit is contained in:
William Falcon 2020-08-20 09:01:40 -04:00 committed by GitHub
parent d51155c7ad
commit da49b15e19
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/callbacks/model_checkpoint.py
View File

@ -314,7 +314,7 @@ class ModelCheckpoint(Callback):
def __warn_deprecated_monitor_key(self): def __warn_deprecated_monitor_key(self):
using_result_obj = os.environ.get('PL_USING_RESULT_OBJ', None) using_result_obj = os.environ.get('PL_USING_RESULT_OBJ', None)
invalid_key = self.monitor not in ['val_loss', 'checkpoint_on'] invalid_key = self.monitor not in ['val_loss', 'checkpoint_on', 'loss']
if using_result_obj and not self.warned_result_obj and invalid_key: if using_result_obj and not self.warned_result_obj and invalid_key:
self.warned_result_obj = True self.warned_result_obj = True
m = f""" m = f"""