From 23137ea08a8ccec074c4e68294902464c1f36ca6 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 3 Jul 2019 16:38:03 -0400 Subject: [PATCH] added single node distdataparallel --- pytorch_lightning/pt_overrides/override_data_parallel.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pytorch_lightning/pt_overrides/override_data_parallel.py b/pytorch_lightning/pt_overrides/override_data_parallel.py index cf4f476e8a..18bc4162fc 100644 --- a/pytorch_lightning/pt_overrides/override_data_parallel.py +++ b/pytorch_lightning/pt_overrides/override_data_parallel.py @@ -28,6 +28,7 @@ class LightningDataParallel(DataParallel): """ def parallel_apply(self, replicas, inputs, kwargs): + print('parallel_apply called') return parallel_apply(replicas, inputs, kwargs, self.device_ids[:len(replicas)])