You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by ro...@apache.org on 2019/04/16 20:07:35 UTC
[incubator-mxnet] branch master updated: Update
inception_inference.cpp (#14674)
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 <49...@users.noreply.github.com>
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.