Skip to content

Commit

Permalink
Merge pull request #213 from astonzhang/nin
Browse files Browse the repository at this point in the history
consistent indentation
  • Loading branch information
astonzhang authored Feb 28, 2018
2 parents 8e7dff3 + 8836e9b commit 8305cc1
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -59,8 +59,8 @@ def transform_mnist(data, label):

mnist_train = gluon.data.vision.FashionMNIST(root=root, train=True, transform=None)
mnist_test = gluon.data.vision.FashionMNIST(root=root, train=False, transform=None)
# Transform later to avoid memory explosion.
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)
test_data = DataLoader(mnist_test, batch_size, shuffle=False, transform=transform_mnist)
return (train_data, test_data)

Expand Down

0 comments on commit 8305cc1

Please sign in to comment.