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 <roywei...@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(