Change weight to width (#211)

utils.py

@@ -49,7 +49,7 @@ def load_data_fashion_mnist(batch_size, resize=None, root="~/.mxnet/datasets/fas
             for i in range(n):
                 new_data[i] = image.imresize(data[i], resize, resize)
             data = new_data
-        # change data from batch x height x weight x channel to batch x channel x height x weight
+        # change data from batch x height x width x channel to batch x channel x height x width
         return nd.transpose(data.astype('float32'), (0,3,1,2))/255, label.astype('float32')
     mnist_train = gluon.data.vision.FashionMNIST(root=root, train=True, transform=transform_mnist)
     mnist_test = gluon.data.vision.FashionMNIST(root=root, train=False, transform=transform_mnist)