Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #962 from tqchen/master

Browse Source 
Fix continue training in CLI
This commit is contained in:
Tianqi Chen 2016-03-10 22:11:45 -08:00
commit 39359edbd8
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/cli_main.cc
View File

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