code: def ddp_model(c: Configs):
      return DistributedDataParallel(
          Net().to(c.device), device_ids=[c.device]
      )
  
file_path: labml/samples/ddp/mnist.py
imports: import datetime
  import torch.distributed
  from torch.nn.parallel import DistributedDataParallel
  from labml import experiment, monit
  from labml.configs import option
  from labml.logger import inspect
  from pytorch.mnist.e_labml_helpers import Configs, Net
label_desc: Use image node_id 1 for calling a global function with example usage: ddp_model(c) and returns: DistributedDataParallel