MCPcopy Create free account
hub / github.com/VincentHancoder/SSGD / build_ddp

Function build_ddp

mmdet/utils/util_distribution.py:34–59  ·  view source on GitHub ↗

Build DistributedDataParallel module by device type. If device is cuda, return a MMDistributedDataParallel model; if device is mlu, return a MLUDistributedDataParallel model. Args: model (:class:`nn.Module`): module to be parallelized. device (str): device type, mlu or

(model, device='cuda', *args, **kwargs)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 2

mainFunction · 0.90
train_detectorFunction · 0.90

Calls 2

mluMethod · 0.80
cudaMethod · 0.45

Tested by 1

mainFunction · 0.72