[incubator-mxnet] branch master updated: fix train mnist for inception-bn and resnet (#13239)

[ns...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

nswamy pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git


The following commit(s) were added to refs/heads/master by this push:
     new 1ef83c9  fix train mnist for inception-bn and resnet (#13239)
1ef83c9 is described below

commit 1ef83c953ff58bf8c444ad84926a1e84745aa00f
Author: Lai Wei <ro...@gmail.com>
AuthorDate: Thu Nov 15 11:46:15 2018 -0800

    fix train mnist for inception-bn and resnet (#13239)
---
 example/image-classification/train_mnist.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/example/image-classification/train_mnist.py b/example/image-classification/train_mnist.py
index 2bc4289..17a5a37 100644
--- a/example/image-classification/train_mnist.py
+++ b/example/image-classification/train_mnist.py
@@ -72,6 +72,7 @@ if __name__ == '__main__':
                         help='the number of training examples')
 
     parser.add_argument('--add_stn',  action="store_true", default=False, help='Add Spatial Transformer Network Layer (lenet only)')
+    parser.add_argument('--image_shape', default='1, 28, 28', help='shape of training images')
 
     fit.add_fit_args(parser)
     parser.set_defaults(