From 8836e9b32ece4e762d4e450ed8053e1517e3f3ed Mon Sep 17 00:00:00 2001 From: Aston Zhang Date: Wed, 28 Feb 2018 00:50:54 +0000 Subject: [PATCH] consistent indentation --- utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils.py b/utils.py index 2086affc1..5c9eb83f1 100644 --- a/utils.py +++ b/utils.py @@ -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)