diff --git a/src/cli_main.cc b/src/cli_main.cc index 039ce4070..f85202aef 100644 --- a/src/cli_main.cc +++ b/src/cli_main.cc @@ -297,6 +297,7 @@ void CLIPredict(const CLIParam& param) { std::unique_ptr learner(Learner::Create({})); std::unique_ptr fi( dmlc::Stream::Create(param.model_in.c_str(), "r")); + learner->Configure(param.cfg); learner->Load(fi.get()); if (param.silent == 0) {