[GitHub] juliensimon opened a new issue #8969: A couple of issues in benchmark_score.py

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

juliensimon opened a new issue #8969: A couple of issues in benchmark_score.py
URL: https://github.com/apache/incubator-mxnet/issues/8969
 
 
   Here are a couple of issues in incubator-mxnet/example/image-classification/benchmark_score.py.
   
   Both prevent the script from running.
   
   --> Issue #1
   def get_symbol(network, batch_size):
       image_shape = (3,299,299) if network == 'inception-v3' else (3,224,224)
       num_layers = 0
   
   get_symbol() for vgg16 doesn't accept a zero value for the number of layers.
   Quick fix: insert this after "num_layers = 0"
   if network=='vgg':
       num_layers=16
   Or use the same solution as for resnet (split and extract number of layers)
   
   --> Issue #2
   devs = [mx.gpu(0)] if len(get_gpus()) > 0 else []
   
   get_gpus() relies on nvidia-smi which is not installed on machines that don't have GPUs (say, c4 instances on AWS).
   

----------------------------------------------------------------
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