From 2ccbcc7e7949bd1950bef420541f41a94626ba4b Mon Sep 17 00:00:00 2001 From: Dariush Wahdany <86673488+lsc64@users.noreply.github.com> Date: Thu, 5 Oct 2023 23:59:31 +0200 Subject: [PATCH] Update examples/mnist_lightning.py Co-authored-by: Karthik Prasad --- examples/mnist_lightning.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/mnist_lightning.py b/examples/mnist_lightning.py index 2155feb0..9313dbcd 100644 --- a/examples/mnist_lightning.py +++ b/examples/mnist_lightning.py @@ -100,7 +100,7 @@ def configure_optimizers(self): if self.enable_dp: self.trainer.fit_loop.setup_data() - dataloader = self.trainer.train_dataloader + data_loader = self.trainer.train_dataloader # transform (model, optimizer, dataloader) to DP-versions if hasattr(self, "dp"):