Skip to content

Commit

Permalink
remove debug info
Browse files Browse the repository at this point in the history
  • Loading branch information
wbo4958 committed Jul 20, 2020
1 parent 633baed commit fd6eb08
Showing 1 changed file with 0 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -354,7 +354,6 @@ object XGBoost extends Serializable {
// TODO should we throw exception ?
logger.warn("XGBoost only supports 1 gpu per worker")
}
logger.warn("===================================xxxxxxxxxxxx")
// take the first one
addrs.head.toInt
} else {
Expand Down Expand Up @@ -402,7 +401,6 @@ object XGBoost extends Serializable {
logger.info("Leveraging gpu device " + gpuId + " to train")
params = params + ("gpu_id" -> gpuId)
}
logger.info(params)
val booster = if (makeCheckpoint) {
SXGBoost.trainAndSaveCheckpoint(
watches.toMap("train"), params, numRounds,
Expand Down

0 comments on commit fd6eb08

Please sign in to comment.