updated args

2019-06-25 19:45:31 -04:00 · 2019-06-25 19:45:31 -04:00 · 158aca26e2
parent cf57be9dca
commit 158aca26e2
1 changed files with 2 additions and 0 deletions
--- a/pytorch_lightning/pt_overrides/override_data_parallel.py
+++ b/pytorch_lightning/pt_overrides/override_data_parallel.py
@ -1,5 +1,6 @@
 from itertools import chain
 from torch.nn import DataParallel
+import pdb


 class LightningDataParallel(DataParallel):
@ -8,6 +9,7 @@ class LightningDataParallel(DataParallel):
    """

    def forward(self, *inputs, **kwargs):
+        pdb.set_trace()
        if not self.device_ids:
            # -------------
            # MAIN CHANGE