We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 8e7dff3 + 8836e9b commit 8305cc1Copy full SHA for 8305cc1
utils.py
@@ -59,8 +59,8 @@ def transform_mnist(data, label):
59
60
mnist_train = gluon.data.vision.FashionMNIST(root=root, train=True, transform=None)
61
mnist_test = gluon.data.vision.FashionMNIST(root=root, train=False, transform=None)
62
- # Transform later to avoid memory explosion.
63
- train_data = DataLoader(mnist_train, batch_size, shuffle=True, transform=transform_mnist)
+ # Transform later to avoid memory explosion.
+ train_data = DataLoader(mnist_train, batch_size, shuffle=True, transform=transform_mnist)
64
test_data = DataLoader(mnist_test, batch_size, shuffle=False, transform=transform_mnist)
65
return (train_data, test_data)
66
0 commit comments