Fix bug in MergeModel.cpp. (#5605)

PaddlePaddle · Nov 15, 2017 · 9f28925 · 9f28925
1 parent d3ef254
commit 9f28925
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/paddle/trainer/MergeModel.cpp b/paddle/trainer/MergeModel.cpp
@@ -27,16 +27,16 @@ using namespace paddle;  // NOLINT
 using namespace std;     // NOLINT
 
 int main(int argc, char** argv) {
+  initMain(argc, argv);
+  initPython(argc, argv);
+
   if (FLAGS_model_dir.empty() || FLAGS_config_file.empty() ||
       FLAGS_model_file.empty()) {
     LOG(INFO) << "Usage: ./paddle_merge_model --model_dir=pass-00000 "
                  "--config_file=config.py --model_file=out.paddle";
     return 0;
   }
 
-  initMain(argc, argv);
-  initPython(argc, argv);
-
   string confFile = FLAGS_config_file;
 #ifndef PADDLE_WITH_CUDA
   FLAGS_use_gpu = false;