From aa360e9adbfad31786cf1ec7b2f241192f72a168 Mon Sep 17 00:00:00 2001 From: Jiaxin Shan Date: Wed, 24 Mar 2021 16:13:54 -0700 Subject: [PATCH] Change mnist example to use FashionMNIST --- examples/mnist/mnist.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/mnist/mnist.py b/examples/mnist/mnist.py index a7d736cab..74eb0f803 100644 --- a/examples/mnist/mnist.py +++ b/examples/mnist/mnist.py @@ -117,14 +117,14 @@ def main(): kwargs = {'num_workers': 1, 'pin_memory': True} if use_cuda else {} train_loader = torch.utils.data.DataLoader( - datasets.MNIST('../data', train=True, download=True, + datasets.FashionMNIST('../data', train=True, download=True, transform=transforms.Compose([ transforms.ToTensor(), transforms.Normalize((0.1307,), (0.3081,)) ])), batch_size=args.batch_size, shuffle=True, **kwargs) test_loader = torch.utils.data.DataLoader( - datasets.MNIST('../data', train=False, transform=transforms.Compose([ + datasets.FashionMNIST('../data', train=False, transform=transforms.Compose([ transforms.ToTensor(), transforms.Normalize((0.1307,), (0.3081,)) ])),