fixed bad warn for result obj (#3072)

2020-08-20 09:01:40 -04:00 · 2020-08-20 09:01:40 -04:00 · da49b15e19
parent d51155c7ad
commit da49b15e19
1 changed files with 1 additions and 1 deletions
--- a/pytorch_lightning/callbacks/model_checkpoint.py
+++ b/pytorch_lightning/callbacks/model_checkpoint.py
@ -314,7 +314,7 @@ class ModelCheckpoint(Callback):

    def __warn_deprecated_monitor_key(self):
        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:
            self.warned_result_obj = True
            m = f"""