Skip to content

Commit

Permalink
consistent indentation
Browse files Browse the repository at this point in the history
  • Loading branch information
astonzhang committed Feb 28, 2018
1 parent f120361 commit 8836e9b
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 8836e9b

Please sign in to comment.