gpu_device_id and gpu_platform_id

mlr3learners / mlr3learners.lightgbm

Learners from {lightgbm} for mlr3

GNU Lesser General Public License v2.1

9 stars 3 forks source link

Open asheetal opened 3 years ago

asheetal commented 3 years ago

lightgbm provides parameters on its documentation

gpu_platform_id = 0
gpu_device_id = 0

how do pass them to lightgbm from the learner?