Update examples/mnist_lightning.py

Co-authored-by: Karthik Prasad <[email protected]>
pytorch · Jan 5, 2024 · 2ccbcc7 · 2ccbcc7
1 parent ad26433
commit 2ccbcc7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"):