Skip to content
This repository was archived by the owner on Sep 19, 2022. It is now read-only.

Commit 2aae331

Browse files
authored
Change mnist example to use FashionMNIST (#327)
1 parent 4aeb650 commit 2aae331

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/mnist/mnist.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -117,14 +117,14 @@ def main():
117117

118118
kwargs = {'num_workers': 1, 'pin_memory': True} if use_cuda else {}
119119
train_loader = torch.utils.data.DataLoader(
120-
datasets.MNIST('../data', train=True, download=True,
120+
datasets.FashionMNIST('../data', train=True, download=True,
121121
transform=transforms.Compose([
122122
transforms.ToTensor(),
123123
transforms.Normalize((0.1307,), (0.3081,))
124124
])),
125125
batch_size=args.batch_size, shuffle=True, **kwargs)
126126
test_loader = torch.utils.data.DataLoader(
127-
datasets.MNIST('../data', train=False, transform=transforms.Compose([
127+
datasets.FashionMNIST('../data', train=False, transform=transforms.Compose([
128128
transforms.ToTensor(),
129129
transforms.Normalize((0.1307,), (0.3081,))
130130
])),

0 commit comments

Comments
 (0)