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 2019/03/05 19:36:46 UTC
[GitHub] [incubator-mxnet] roywei edited a comment on issue #14312:
[MXNET-1332]predict on gpu
roywei edited a comment on issue #14312: [MXNET-1332]predict on gpu
URL: https://github.com/apache/incubator-mxnet/pull/14312#issuecomment-469812636
@szha Hi, so when users load a model on GPU context, and load an image & apply pre-processing, it's very easy to forget to copy data to GPU. Then this error message will be thrown.
I m trying to help the user copy the data for signle device use cases. Do you think it's better to just error out at module.predict() level with a clear error message? Let the user handle all the context.
The error is actually from : https://github.com/apache/incubator-mxnet/blob/master/python/mxnet/module/executor_group.py#L65
data is `<type 'numpy.uint8'>`
while targets is `<type 'numpy.float32'>`
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to 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