From 59d59a968d5dd3edf0b378d7f56c1d63a709e4d2 Mon Sep 17 00:00:00 2001
From: tqchen <tianqi.tchen@gmail.com>
Date: Thu, 10 Mar 2016 19:39:09 -0800
Subject: [PATCH] Fix continue training in CLI

---
 src/cli_main.cc | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/cli_main.cc b/src/cli_main.cc
index 9ece093da..039ce4070 100644
--- a/src/cli_main.cc
+++ b/src/cli_main.cc
@@ -174,7 +174,6 @@ void CLITrain(const CLIParam& param) {
   }
   // initialize the learner.
   std::unique_ptr<Learner> learner(Learner::Create(cache_mats));
-  learner->Configure(param.cfg);
   int version = rabit::LoadCheckPoint(learner.get());
   if (version == 0) {
     // initializ the model if needed.
@@ -182,7 +181,9 @@ void CLITrain(const CLIParam& param) {
       std::unique_ptr<dmlc::Stream> fi(
           dmlc::Stream::Create(param.model_in.c_str(), "r"));
       learner->Load(fi.get());
+      learner->Configure(param.cfg);
     } else {
+      learner->Configure(param.cfg);
       learner->InitModel();
     }
   }