You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@mxnet.apache.org by GitBox <gi...@apache.org> on 2018/04/24 02:08:51 UTC

[GitHub] juliusshufan commented on a change in pull request #10646: Correctly set the device context to CPU at symbolic mode when --gpus not set

juliusshufan commented on a change in pull request #10646: Correctly set the device context to CPU at symbolic mode when --gpus not set
URL: https://github.com/apache/incubator-mxnet/pull/10646#discussion_r183587582
 
 

 ##########
 File path: example/gluon/image_classification.py
 ##########
 @@ -255,7 +255,10 @@ def main():
         data = mx.sym.var('data')
         out = net(data)
         softmax = mx.sym.SoftmaxOutput(out, name='softmax')
-        mod = mx.mod.Module(softmax, context=[mx.gpu(i) for i in range(num_gpus)] if num_gpus > 0 else [mx.cpu()])
 
 Review comment:
   @piiswrong , thanks for review and you are right, commit revised. Thanks. 

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