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"):