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)])