Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
3,888 Commits 32 Branches 59 Tags
Commit Graph

1 Commits

Author SHA1 Message Date
Jiaming Yuan
c589eff941
De-duplicate GPU parameters. (#4454) 
* Only define `gpu_id` and `n_gpus` in `LearnerTrainParam`
* Pass LearnerTrainParam through XGBoost vid factory method.
* Disable all GPU usage when GPU related parameters are not specified (fixes XGBoost choosing GPU over aggressively).
* Test learner train param io.
* Fix gpu pickling.
 2019-05-29 11:55:57 +08:00