From 4db32984c600a07fadfd164a50c0ad0a9c75cdb0 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Thu, 18 Jul 2019 11:39:13 -0400 Subject: [PATCH] set dp as default backend --- pytorch_lightning/pt_overrides/override_data_parallel.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/pytorch_lightning/pt_overrides/override_data_parallel.py b/pytorch_lightning/pt_overrides/override_data_parallel.py index 2366b882..67188b05 100644 --- a/pytorch_lightning/pt_overrides/override_data_parallel.py +++ b/pytorch_lightning/pt_overrides/override_data_parallel.py @@ -67,7 +67,6 @@ class LightningDataParallel(DataParallel): def parallel_apply(self, replicas, inputs, kwargs): - print('LDP') return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)]) @@ -80,7 +79,6 @@ class LightningDistributedDataParallel(DistributedDataParallel): return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)]) def forward(self, *inputs, **kwargs): - print('LDDP') self._sync_params() if self.device_ids: inputs, kwargs = self.scatter(inputs, kwargs, self.device_ids)