[GPU-Plugin] Make node_idx type 32 bit for hist algo. Set default n_gpus to 1. (#2445)

This commit is contained in:
Rory Mitchell
2017-06-23 18:26:45 +12:00
committed by GitHub
parent 169c983b5f
commit 0e48f87529
3 changed files with 4 additions and 4 deletions

View File

@@ -196,7 +196,7 @@ struct TrainParam : public dmlc::Parameter<TrainParam> {
.describe("gpu to use for single gpu algorithms");
DMLC_DECLARE_FIELD(n_gpus)
.set_lower_bound(-1)
.set_default(-1)
.set_default(1)
.describe("Number of GPUs to use for multi-gpu algorithms: -1=use all GPUs");
// add alias of parameters
DMLC_DECLARE_ALIAS(reg_lambda, lambda);