From ad264330f060996e791bcf2c2781a7d3cfd1a361 Mon Sep 17 00:00:00 2001 From: Dariush Wahdany Date: Mon, 11 Sep 2023 15:29:17 +0200 Subject: [PATCH] feat: update lightning example to lightning 2.0 --- examples/mnist_lightning.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/examples/mnist_lightning.py b/examples/mnist_lightning.py index 42d93a3d..2155feb0 100644 --- a/examples/mnist_lightning.py +++ b/examples/mnist_lightning.py @@ -99,11 +99,8 @@ def configure_optimizers(self): optimizer = optim.SGD(self.parameters(), lr=self.lr, momentum=0) if self.enable_dp: - data_loader = ( - # soon there will be a fancy way to access train dataloader, - # see https://github.com/PyTorchLightning/pytorch-lightning/issues/10430 - self.trainer._data_connector._train_dataloader_source.dataloader() - ) + self.trainer.fit_loop.setup_data() + dataloader = self.trainer.train_dataloader # transform (model, optimizer, dataloader) to DP-versions if hasattr(self, "dp"):