From 22bb168d8d16ff694585ba803439323c34b8c76c Mon Sep 17 00:00:00 2001 From: svohara Date: Mon, 11 Jan 2016 12:36:02 -0700 Subject: [PATCH] Fix bug preventing train_mnist.py from working when data-dir is an S3 uri. --- example/image-classification/train_mnist.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/example/image-classification/train_mnist.py b/example/image-classification/train_mnist.py index d8751833a1e7..f7fb07777d79 100644 --- 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",