From f20839bf81677e0ed2169bb8e5fcbeb920dbc5a6 Mon Sep 17 00:00:00 2001
From: Bin Long <sjtubinlong@users.noreply.github.com>
Date: Wed, 16 Oct 2019 15:54:08 +0800
Subject: [PATCH] Update seg_predictor.cpp

---
 inference/predictor/seg_predictor.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/inference/predictor/seg_predictor.cpp b/inference/predictor/seg_predictor.cpp
index d70084f6..c6dae1e4 100644
--- a/inference/predictor/seg_predictor.cpp
+++ b/inference/predictor/seg_predictor.cpp
@@ -43,6 +43,8 @@ namespace PaddleSolution {
                 auto param_file = utils::path_join(model_dir, params_filename);
                 config.SetModel(prog_file, param_file);
                 config.SwitchUseFeedFetchOps(false);
+                config.SwitchSpecifyInputNames(true);
+                config.EnableMemoryOptim();
                 _main_predictor = paddle::CreatePaddlePredictor(config);
             }
             else {
-- 
GitLab