diff --git a/src/pytorch_lightning/callbacks/early_stopping.py b/src/pytorch_lightning/callbacks/early_stopping.py index 6c1a43e1d1..4acff87dfa 100644 --- a/src/pytorch_lightning/callbacks/early_stopping.py +++ b/src/pytorch_lightning/callbacks/early_stopping.py @@ -261,7 +261,7 @@ class EarlyStopping(Callback): @staticmethod def _log_info(trainer: Optional["pl.Trainer"], message: str, log_rank_zero_only: bool) -> None: - rank = _get_rank(trainer) + rank = _get_rank(strategy=(trainer.strategy if trainer is not None else None)) if trainer is not None and trainer.world_size <= 1: rank = None message = rank_prefixed_message(message, rank)