get_device_type¶

dragon.vm.dali.get_device_type(mixed=False)[source]¶

Return the current nesting device type.

Parameters:

• mixed (bool, optional, default=False) – True to return mixed for gpu device.

Returns:

{‘cpu’, ‘gpu’, ‘mixed’} – The current device type.