Minor warning message fix (Lightning-AI#2173)

JanSellner · web-flow · commit ea6b350b32d3 · 2020-06-13T10:02:15.000-04:00
diff --git a/pytorch_lightning/callbacks/model_checkpoint.py b/pytorch_lightning/callbacks/model_checkpoint.py
@@ -260,7 +260,7 @@ def on_validation_end(self, trainer, pl_module):
             if not isinstance(current, torch.Tensor):
                 rank_zero_warn(
                     f'The metric you returned {current} must be a Torch.Tensor instance, checkpoint not saved '
-                    f'HINT: what is the value of {self.monitor} in validation_end()?', RuntimeWarning
+                    f'HINT: what is the value of {self.monitor} in validation_epoch_end()?', RuntimeWarning
                 )
 
             if current is None:

Original file line number	Diff line number	Diff line change
`@@ -260,7 +260,7 @@ def on_validation_end(self, trainer, pl_module):`
`260`	`260`	`if not isinstance(current, torch.Tensor):`
`261`	`261`	`rank_zero_warn(`
`262`	`262`	`f'The metric you returned {current} must be a Torch.Tensor instance, checkpoint not saved '`
`263`		`- f'HINT: what is the value of {self.monitor} in validation_end()?', RuntimeWarning`
	`263`	`+ f'HINT: what is the value of {self.monitor} in validation_epoch_end()?', RuntimeWarning`
`264`	`264`	`)`
`265`	`265`
`266`	`266`	`if current is None:`