cerebras.modelzoo.data.vision.utils.is_gpu_distributed#

cerebras.modelzoo.data.vision.utils.is_gpu_distributed()[source]#

Returns True if DDP is enabled

previous

cerebras.modelzoo.data.vision.utils.create_worker_cache

next

cerebras.modelzoo.data.vision.utils.num_tasks