fix: MNIST minor bug (Lightning-AI#5075)

Co-authored-by: Roger Shieh <[email protected]>
gianscarpe · Dec 12, 2020 · d38e4d1 · d38e4d1
1 parent 5f34f2b
commit d38e4d1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/source/introduction_guide.rst b/docs/source/introduction_guide.rst
@@ -601,8 +601,8 @@ In this method we do all the preparation we need to do once (instead of on every
         def setup(self, stage):
             # transform
             transform=transforms.Compose([transforms.ToTensor()])
-            MNIST(os.getcwd(), train=True, download=False, transform=transform)
-            MNIST(os.getcwd(), train=False, download=False, transform=transform)
+            mnist_train = MNIST(os.getcwd(), train=True, download=False, transform=transform)
+            mnist_test = MNIST(os.getcwd(), train=False, download=False, transform=transform)
 
             # train/val split
             mnist_train, mnist_val = random_split(mnist_train, [55000, 5000])