Source code for super_gradients.common.data_types.enum.multi_gpu_mode
from enum import Enum
[docs]class MultiGPUMode(str, Enum):
"""
MultiGPUMode
Attributes:
OFF - Single GPU Mode / CPU Mode
DATA_PARALLEL - Multiple GPUs, Synchronous
DISTRIBUTED_DATA_PARALLEL - Multiple GPUs, Asynchronous
"""
OFF = 'Off'
DATA_PARALLEL = 'DP'
DISTRIBUTED_DATA_PARALLEL = 'DDP'
AUTO = "AUTO"