diff --git a/pytorch_lightning/models/trainer.py b/pytorch_lightning/models/trainer.py index 62df9ef4f9..74b147e3d2 100644 --- a/pytorch_lightning/models/trainer.py +++ b/pytorch_lightning/models/trainer.py @@ -19,7 +19,7 @@ import tqdm from pytorch_lightning.root_module.memory import get_gpu_memory_map from pytorch_lightning.root_module.model_saving import TrainerIO from pytorch_lightning.pt_overrides.override_data_parallel import LightningDistributedDataParallel, LightningDataParallel - +from pytorch_lightning.utils.debugging import ForkedPdb try: from apex import amp @@ -254,8 +254,7 @@ class Trainer(TrainerIO): @property def __tng_tqdm_dic(self): - import pdb - pdb.set_trace() + ForkedPdb().set_trace() tqdm_dic = { 'tng_loss': '{0:.3f}'.format(self.avg_loss), 'v_nb': '{}'.format(self.experiment.version), diff --git a/tests/test_models.py b/tests/test_models.py index 580ba1afda..a02bde91a7 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -242,6 +242,5 @@ def test_amp_gpu_ddp(): - if __name__ == '__main__': pytest.main([__file__])