fix bug in loading config for pred task (dmlc#2795)

viviancodes · Oct 20, 2017 · c09ad42 · c09ad42
1 parent 5dca674
commit c09ad42
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cli_main.cc b/src/cli_main.cc
@@ -318,8 +318,8 @@ void CLIPredict(const CLIParam& param) {
   std::unique_ptr<Learner> learner(Learner::Create({}));
   std::unique_ptr<dmlc::Stream> fi(
       dmlc::Stream::Create(param.model_in.c_str(), "r"));
-  learner->Configure(param.cfg);
   learner->Load(fi.get());
+  learner->Configure(param.cfg);
 
   if (param.silent == 0) {
     LOG(CONSOLE) << "start prediction...";