From 59fcabf53a72b23e6efa0a65f58c79cd9bff0b1a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= Date: Wed, 7 Sep 2022 22:56:39 +0200 Subject: [PATCH] Refactor `_get_rank` utility to take strategy instead of trainer (#14546) --- src/pytorch_lightning/callbacks/early_stopping.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)