From 2181ad1bc737880c3949d43735101033d126baae Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 27 Apr 2020 07:12:04 -0400 Subject: [PATCH] ddp pickle --- pytorch_lightning/trainer/data_loading.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pytorch_lightning/trainer/data_loading.py b/pytorch_lightning/trainer/data_loading.py index 7d0b8180..92fb73e2 100644 --- a/pytorch_lightning/trainer/data_loading.py +++ b/pytorch_lightning/trainer/data_loading.py @@ -134,6 +134,7 @@ class TrainerDataLoadingMixin(ABC): 'ddp': self.num_nodes * self.num_processes, 'ddp2': self.num_nodes, } + import pdb; pdb.set_trace() sampler = DistributedSampler( dataloader.dataset, num_replicas=world_size.get(self.distributed_backend, 0),