pprint debugging

Lightning-AI · Jul 27, 2022 · 5a51c5a · 5a51c5a
1 parent 305730d
commit 5a51c5a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/pytorch_lightning/strategies/ipu.py b/src/pytorch_lightning/strategies/ipu.py
@@ -234,7 +234,13 @@ def _convert_to_poptorch_loader(
             dataloader, sampler, mode, self.replication_factor > 1
         )
         opts = self.training_opts if mode == RunningStage.TRAINING else self.inference_opts
+        from pprint import pprint
+
+        pprint(opts)
+        pprint(dl_args)
+        pprint(dl_kwargs)
         dataloader = poptorch.DataLoader(opts, *dl_args, **dl_kwargs)
+        pprint(dataloader)
         return dataloader
 
     def _handle_gradient_accumulation_steps(self) -> None: