[GPU-Plugin] Make node_idx type 32 bit for hist algo. Set default n_gpus to 1. (#2445)
This commit is contained in:
@@ -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);
|
||||
|
||||
Reference in New Issue
Block a user