From d873ea5225bd0cdb55437dd5cbfc5577b36f12f1 Mon Sep 17 00:00:00 2001 From: huzhiqiang <912790387@qq.com> Date: Wed, 29 Apr 2020 10:19:48 +0800 Subject: [PATCH] [BUG FIX][X86 Demo] Fix the issue that x86 light_api_demo can not operate successfully #3529 --- .../cxx/x86_mobilenetv1_light_demo/mobilenet_light_api.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lite/demo/cxx/x86_mobilenetv1_light_demo/mobilenet_light_api.cc b/lite/demo/cxx/x86_mobilenetv1_light_demo/mobilenet_light_api.cc index 17997a228f..763a3fe887 100644 --- a/lite/demo/cxx/x86_mobilenetv1_light_demo/mobilenet_light_api.cc +++ b/lite/demo/cxx/x86_mobilenetv1_light_demo/mobilenet_light_api.cc @@ -24,10 +24,10 @@ int64_t ShapeProduction(const shape_t& shape) { return res; } -void RunModel(std::string model_dir) { +void RunModel(std::string model_name) { // 1. Create MobileConfig MobileConfig config; - config.set_model_dir(model_dir); + config.set_model_from_file(model_name); // 2. Create PaddlePredictor by CxxConfig std::shared_ptr predictor = CreatePaddlePredictor(config); -- GitLab