Fix issue dmlc#1236: cli_main crashes when dumping count:poisson model

yangs16 · Jun 6, 2016 · ef2b319 · ef2b319
1 parent 9a48a40
commit ef2b319
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/dmlc-core b/dmlc-core
diff --git a/src/cli_main.cc b/src/cli_main.cc
@@ -271,6 +271,7 @@ void CLIDump2Text(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());
   // dump data
   std::vector<std::string> dump = learner->Dump2Text(fmap, param.dump_stats);
+2 −1		CMakeLists.txt
+17 −0		include/dmlc/base.h
+3 −2		include/dmlc/io.h
+0 −10		include/dmlc/logging.h
+10 −3		include/dmlc/timer.h
+12 −11		scripts/setup_nvcc.sh
+5 −5		scripts/travis/travis_setup_env.sh
+1 −1		src/io/input_split_base.cc
+14 −9		tracker/dmlc_tracker/mpi.py
+8 −5		tracker/dmlc_tracker/opts.py
+76 −0		tracker/dmlc_tracker/ssh.py
+9 −8		tracker/dmlc_tracker/tracker.py