Skip to content

Commit d38e4d1

Browse files
fix: MNIST minor bug (#5075)
Co-authored-by: Roger Shieh <[email protected]>
1 parent 5f34f2b commit d38e4d1

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/source/introduction_guide.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -601,8 +601,8 @@ In this method we do all the preparation we need to do once (instead of on every
601601
def setup(self, stage):
602602
# transform
603603
transform=transforms.Compose([transforms.ToTensor()])
604-
MNIST(os.getcwd(), train=True, download=False, transform=transform)
605-
MNIST(os.getcwd(), train=False, download=False, transform=transform)
604+
mnist_train = MNIST(os.getcwd(), train=True, download=False, transform=transform)
605+
mnist_test = MNIST(os.getcwd(), train=False, download=False, transform=transform)
606606
607607
# train/val split
608608
mnist_train, mnist_val = random_split(mnist_train, [55000, 5000])

0 commit comments

Comments
 (0)