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.