Skip to content

Commit 9f28925

Browse files
authored
Fix bug in MergeModel.cpp. (#5605)
1 parent d3ef254 commit 9f28925

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

paddle/trainer/MergeModel.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -27,16 +27,16 @@ using namespace paddle; // NOLINT
2727
using namespace std; // NOLINT
2828

2929
int main(int argc, char** argv) {
30+
initMain(argc, argv);
31+
initPython(argc, argv);
32+
3033
if (FLAGS_model_dir.empty() || FLAGS_config_file.empty() ||
3134
FLAGS_model_file.empty()) {
3235
LOG(INFO) << "Usage: ./paddle_merge_model --model_dir=pass-00000 "
3336
"--config_file=config.py --model_file=out.paddle";
3437
return 0;
3538
}
3639

37-
initMain(argc, argv);
38-
initPython(argc, argv);
39-
4040
string confFile = FLAGS_config_file;
4141
#ifndef PADDLE_WITH_CUDA
4242
FLAGS_use_gpu = false;

0 commit comments

Comments
 (0)