This is an automated email from the ASF dual-hosted git repository.
roshrini 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 8e04b88 Update inception_inference.cpp (#14674)
8e04b88 is described below
commit 8e04b8870c01e2c7e88511aec5c7baa668c69a61
Author: champagne828 <[email protected]>
AuthorDate: Wed Apr 17 04:07:07 2019 +0800
Update inception_inference.cpp (#14674)
fix a bug when use --gpu
---
cpp-package/example/inference/inception_inference.cpp | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/cpp-package/example/inference/inception_inference.cpp
b/cpp-package/example/inference/inception_inference.cpp
index 78487e6..fa56001 100644
--- a/cpp-package/example/inference/inception_inference.cpp
+++ b/cpp-package/example/inference/inception_inference.cpp
@@ -301,7 +301,7 @@ void Predictor::PredictImage(const std::string& image_file)
{
executor->Forward(false);
// The output is available in executor->outputs.
- auto array = executor->outputs[0].Copy(global_ctx);
+ auto array = executor->outputs[0].Copy(Context::cpu());
/*
* Find out the maximum accuracy and the index associated with that accuracy.
