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

Function build_dp

mmdet/utils/util_distribution.py:10–31  ·  view source on GitHub ↗

build DataParallel module by device type. if device is cuda, return a MMDataParallel model; if device is mlu, return a MLUDataParallel model. Args: model (:class:`nn.Module`): model to be parallelized. device (str): device type, cuda, cpu or mlu. Defaults to cuda.

(model, device='cuda', dim=0, *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