ddp fix

2020-04-27 08:28:39 -04:00 · 2020-04-27 08:28:39 -04:00 · 63addd091c
parent a24c88ab08
commit 63addd091c
1 changed files with 0 additions and 1 deletions
--- a/pytorch_lightning/trainer/distrib_data_parallel.py
+++ b/pytorch_lightning/trainer/distrib_data_parallel.py
@ -378,7 +378,6 @@ class TrainerDDPMixin(ABC):
        :param model:
        :return:
        """
-        import pdb; pdb.set_trace()
        if self.proc_rank == 0:
            path = os.path.join(self.default_root_dir, '__temp_weight_ddp_end.ckpt')
            self.save_checkpoint(path)