From 5a51c5aba9515bc64e7f959f4bbefd6b525a017d Mon Sep 17 00:00:00 2001 From: otaj Date: Wed, 27 Jul 2022 15:19:38 +0200 Subject: [PATCH] pprint debugging --- src/pytorch_lightning/strategies/ipu.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/pytorch_lightning/strategies/ipu.py b/src/pytorch_lightning/strategies/ipu.py index 001ad77fbb5cc..108c725d1abdf 100644 --- 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: