code: def __init__(self):
      super(Model, self).__init__()
      self.a = Parameter(torch.cuda.FloatTensor(4096 * 4096).fill_(1.0))
      self.b = Parameter(torch.cuda.FloatTensor(4096 * 4096).fill_(2.0))
  
file_path: apex/tests/distributed/DDP/ddp_race_condition_test.py
imports: import torch
  import torch.distributed
  from torch.nn import Parameter
  from torch.nn import Module
  from apex.parallel import DistributedDataParallel
  import argparse
  import os
label_desc: Use image node_id 1 to create a new Model object from inherited base classes: Module with example: obj = Model()