From 816da57f30e41e62d5c7880a0e705971759f9eeb Mon Sep 17 00:00:00 2001
From: xzl <zlx_hg@163.com>
Date: Thu, 28 Sep 2017 14:48:39 +0800
Subject: [PATCH] refine paddle_merge_model

---
 paddle/trainer/MergeModel.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/paddle/trainer/MergeModel.cpp b/paddle/trainer/MergeModel.cpp
index 91d89b61a32..18ae6cc9382 100644
--- a/paddle/trainer/MergeModel.cpp
+++ b/paddle/trainer/MergeModel.cpp
@@ -20,6 +20,7 @@ limitations under the License. */
 #include "paddle/utils/PythonUtil.h"
 
 DEFINE_string(model_dir, "", "Directory for separated model files");
+DEFINE_string(config_file, "", "Config file for the model");
 DEFINE_string(model_file, "", "File for merged model file");
 
 using namespace paddle;  // NOLINT
@@ -28,7 +29,7 @@ using namespace std;     // NOLINT
 int main(int argc, char** argv) {
   initMain(argc, argv);
   initPython(argc, argv);
-  string confFile = TrainerConfigHelper::getConfigNameFromPath(FLAGS_model_dir);
+  string confFile = FLAGS_config_file;
 #ifdef PADDLE_ONLY_CPU
   FLAGS_use_gpu = false;
 #endif
-- 
GitLab