[GitHub] roywei opened a new issue #13238: train_mnist.py in example/image-classification does not work with inception-bn and resnet

[GitBox <gi...@apache.org>]

roywei opened a new issue #13238: train_mnist.py in example/image-classification does not work with inception-bn and resnet
URL: https://github.com/apache/incubator-mxnet/issues/13238
 
 
   ## Description
   train_mnist.py in example/image-classification does not work with inception-bn and resnet
   As in title, the following commands will fail: 
   
   ```
   python train_mnist.py --network inception-bn --gpus 0,1,2,3
   ```
   ```
   python train_mnist.py --network resnet --num-layers 110  --gpus 0,1,2,3
   ```
   
   ## Error Message:
   ```
   Traceback (most recent call last):
     File "train_mnist.py", line 93, in <module>
       sym = net.get_symbol(**vars(args))
   TypeError: get_symbol() takes at least 3 arguments (3 given)
   ```
   
   ## Root cause
   The `get_symbol()`  in those models requires additional `image_shape` argument, and `train_mnist.py` does not provide it

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services