fix

Author: Borda

pytorch_lightning/trainer/connectors/logger_connector/logger_connector.py

@@ -38,7 +38,7 @@ def __init__(self, trainer):
         self._logged_metrics = MetricsHolder()
         self._progress_bar_metrics = MetricsHolder()
         self.eval_loop_results = []
-        self._cached_results = {stage: EpochResultStore(trainer, stage) for stage in RunningStage}
+        self._cached_results = {stage.value: EpochResultStore(trainer, stage) for stage in RunningStage}
         self._callback_hook_validator = CallbackHookNameValidator()
         self._current_stage = None
 
@@ -76,7 +76,7 @@ def progress_bar_metrics(self, progress_bar_metrics: Dict) -> None:
 
     @property
     def cached_results(self) -> Union[EpochResultStore, None]:
-        return self._cached_results.get(self._current_stage)    # type: ignore
+        return self._cached_results.get(self._current_stage.value)    # type: ignore
 
     def get_metrics(self, key: str) -> Dict:
         metrics_holder = getattr(self, f"_{key}", None)