Update inception_inference.cpp (apache#14674)

fix a bug when use --gpu
kedarbellare · Apr 20, 2019 · f729926 · f729926
1 parent bbcc269
commit f729926
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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.