ref: finish decoupling apex, LM and backward

Lightning-AI · Oct 10, 2020 · 2f78993 · 2f78993
1 parent 7adbe90
commit 2f78993
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/pytorch_lightning/accelerators/accelerator_connector.py b/pytorch_lightning/accelerators/accelerator_connector.py
@@ -52,7 +52,9 @@ def on_trainer_init(
             self.accelerator = accelerator
             if isinstance(accelerator, Accelerator):
                 self.accelerator.trainer = self
-            distributed_backend = self.accelerator.nickname
+                distributed_backend = self.accelerator.nickname
+            else:
+                distributed_backend = accelerator
 
         self.trainer.deterministic = deterministic
 

diff --git a/pytorch_lightning/utilities/argparse_utils.py b/pytorch_lightning/utilities/argparse_utils.py
@@ -105,9 +105,11 @@ def get_init_arguments_and_types(cls) -> List[Tuple[str, Tuple, Any]]:
         >>> args = get_init_arguments_and_types(Trainer)
         >>> import pprint
         >>> pprint.pprint(sorted(args))  # doctest: +ELLIPSIS +NORMALIZE_WHITESPACE
-        [('accumulate_grad_batches',
-          (<class 'int'>, typing.Dict[int, int], typing.List[list]),
-          1),
+        [('accelerator',
+          (<class 'str'>,
+           <class 'pytorch_lightning.accelerators.base_accelerator.Accelerator'>,
+           <class 'NoneType'>),
+          None),
          ...
          ('callbacks',
           (typing.List[pytorch_lightning.callbacks.base.Callback],