Fix CLI ranking demo. (#6439)

Save model at final round.
2020-11-28 14:12:06 -05:00
parent b0036b339b
commit f4ff1c53fd
3 changed files with 40 additions and 13 deletions
--- a/src/cli_main.cc
+++ b/src/cli_main.cc
@@ -268,7 +268,7 @@ class CLI {
    // always save final round
    if ((param_.save_period == 0 ||
         param_.num_round % param_.save_period != 0) &&
-        param_.model_out != CLIParam::kNull && rabit::GetRank() == 0) {
+        rabit::GetRank() == 0) {
      std::ostringstream os;
      if (param_.model_out == CLIParam::kNull) {
        os << param_.model_dir << '/' << std::setfill('0') << std::setw(4)