Shortcuts

mmengine.device.get_device

mmengine.device.get_device()[source]

Returns the currently existing device type.

Returns

cuda | npu | mlu | mps | cpu.

Return type

str