From 1441f6ee0eeb3f83d4eb2f8a63282a281d0aaba0 Mon Sep 17 00:00:00 2001 From: skhiuk <38586033+Shumpei-Kikuta@users.noreply.github.com> Date: Fri, 11 Dec 2020 12:46:32 +0900 Subject: [PATCH] fix: MNIST minor bug --- docs/source/introduction_guide.rst | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/source/introduction_guide.rst b/docs/source/introduction_guide.rst index d6d082e2ed779..d4cf578e10bda 100644 --- 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])