Fix _call_loggers_load_state_dict

Lightning-AI · May 19, 2022 · 13258e4 · 13258e4
1 parent e14c8b3
commit 13258e4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py
@@ -1720,7 +1720,7 @@ def _call_loggers_load_state_dict(self, checkpoint: Dict[str, Any]) -> None:
  return
 
  for logger in self.loggers:
- state = logger_states.get(logger.state_key, type(logger))
+ state = logger_states.get(logger.state_key)
  if state:
  state = deepcopy(state)
  logger.load_state_dict(state)