Merge pull request apache#1241 from svohara/mnist_example_bug_fix

Fix bug with train_mnist.py when data-dir is an S3 URI
LoneStarAI · Jan 12, 2016 · 08466f8 · 08466f8
2 parents 68d64e5 + 22bb168
commit 08466f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/example/image-classification/train_mnist.py b/example/image-classification/train_mnist.py
@@ -95,7 +95,7 @@ def get_iterator(args, kv):
     data_dir = args.data_dir
     if '://' not in args.data_dir:
         _download(args.data_dir)
-        flat = False if len(data_shape) == 3 else True
+    flat = False if len(data_shape) == 3 else True
 
     train           = mx.io.MNISTIter(
         image       = data_dir + "train-images-idx3-ubyte",