Shortcuts

mmengine.device.get_device

mmengine.device.get_device()[source]

Returns the currently existing device type.

Returns:

cuda | npu | mlu | mps | musa | cpu.

Return type:

str