Directly call TrainingTypePlugin APIs instead of going through the Accelerator

four4fish · four4fish · commit b82d0671c556 · 2021-10-11T17:49:37.000-07:00
diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py
@@ -1025,9 +1025,9 @@ def _run(self, model: "pl.LightningModule") -> Optional[Union[_EVALUATE_OUTPUT,
                                 |                             ||
                          {self._dispatch}                     ||
                                 |                             ||  LIGHTNING
-                  {self.training_type_plugin.start_training}           ||
-                or {self.training_type_plugin.start_evaluating}        ||
-                or {self.training_type_plugin.start_predicting}        ||  FLOW
+         {self.training_type_plugin.start_training}           ||
+       or {self.training_type_plugin.start_evaluating}        ||
+       or {self.training_type_plugin.start_predicting}        ||  FLOW
                                 |                             ||
                          {self.run_stage}                     ||
                                 |                             ||  DIRECTION
