From da49b15e19175e24fbf54c2af0cee25a4a5a0e08 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 20 Aug 2020 09:01:40 -0400 Subject: [PATCH] fixed bad warn for result obj (#3072) --- pytorch_lightning/callbacks/model_checkpoint.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/callbacks/model_checkpoint.py b/pytorch_lightning/callbacks/model_checkpoint.py index f421d6152a..8863051b7d 100644 --- 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"""