ref: part 4 of #3733

Lightning-AI · Oct 2, 2020 · 5d99a75 · 5d99a75
1 parent 9e04ec0
commit 5d99a75
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/pytorch_lightning/trainer/connectors/precision_connector.py b/pytorch_lightning/trainer/connectors/precision_connector.py
@@ -47,7 +47,6 @@ def _setup_amp_backend(self, amp_type: str):
 
         amp_type = amp_type.lower()
         assert amp_type in ('native', 'apex'), f'Unsupported amp type {amp_type}'
-        import pdb; pdb.set_trace()
         if amp_type == 'native':
             if not NATIVE_AMP_AVALAIBLE:
                 rank_zero_warn('You have asked for native AMP but your PyTorch version does not support it.'

diff --git a/tests/models/test_amp.py b/tests/models/test_amp.py
@@ -87,7 +87,6 @@ def test_amp_multi_gpu_ddp_spawn(tmpdir):
 
     model = EvalModelTemplate()
     # tutils.run_model_test(trainer_options, model)
-    import pdb; pdb.set_trace()
     result = trainer.fit(model)
 
     assert result == 1