From 63addd091cfbcc76456c150431fe0ea168bf024c Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 27 Apr 2020 08:28:39 -0400 Subject: [PATCH] ddp fix --- pytorch_lightning/trainer/distrib_data_parallel.py | 1 - 1 file changed, 1 deletion(-) diff --git a/pytorch_lightning/trainer/distrib_data_parallel.py b/pytorch_lightning/trainer/distrib_data_parallel.py index f26901c0ed..659aa7a072 100644 --- 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)