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

4 Commits

Author SHA1 Message Date
Jiaming Yuan
7d93932423
Better message when no GPU is found. (#5594) 2020-04-26 10:00:57 +08:00
Jiaming Yuan
61286c6e8f
Fix wrapping GPU ID and prevent data copying. (#5160) 
* Removed some data copying.

* Make sure gpu_id is valid before any configuration is carried out.
 2019-12-27 16:51:08 +08:00
Jiaming Yuan
3136185bc5
JSON configuration IO. (#5111) 
* Add saving/loading JSON configuration.
* Implement Python pickle interface with new IO routines.
* Basic tests for training continuation.
 2019-12-15 17:31:53 +08:00
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