Processing unit settings

task_typetask_type

Command line: --task-type

DescriptionDescription

The processing unit type to use for training.

Possible values:

• CPU
• GPU

Type

string

Default value

CPU

Supported processing units

CPU and GPU

devicesdevices

Command line: --devices

DescriptionDescription

IDs of the GPU devices to use for training (indices are zero-based).

Format

• <unit ID> for one device (for example, 3)
• <unit ID1>:<unit ID2>:..:<unit IDN> for multiple devices (for example, devices='0:1:3')
• <unit ID1>-<unit IDN> for a range of devices (for example, devices='0-3')

Type

string

Default value

Python package

NULL (all GPU devices are used if the corresponding processing unit type is
selected)

R package

-1 (all GPU devices are used if the corresponding processing unit type is selected)

Command-line

-1 (use all devices)

Supported processing units

CPU