Shortcuts

mmengine.device.get_device

mmengine.device.get_device()[源代码]

Returns the currently existing device type.

返回

cuda | mlu | mps | cpu.

返回类型

str