Skip to content

Commit 2ccbcc7

Browse files
Update examples/mnist_lightning.py
Co-authored-by: Karthik Prasad <prasadkr@uci.edu>
1 parent ad26433 commit 2ccbcc7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/mnist_lightning.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -100,7 +100,7 @@ def configure_optimizers(self):
100100

101101
if self.enable_dp:
102102
self.trainer.fit_loop.setup_data()
103-
dataloader = self.trainer.train_dataloader
103+
data_loader = self.trainer.train_dataloader
104104

105105
# transform (model, optimizer, dataloader) to DP-versions
106106
if hasattr(self, "dp"):

0 commit comments

Comments
 (0)