diff --git a/paddle/trainer/MergeModel.cpp b/paddle/trainer/MergeModel.cpp index 91d89b61a32259b8bbe70fda2579f87ec6b9af00..18ae6cc9382a444d549824c8f8d57f962f17d427 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