code: def __init__(self):
      super(ModelFoo, self).__init__()
      self.linear = torch.nn.Linear(128, 128, bias=False)
      self.loss = torch.nn.MSELoss()
  
file_path: apex/apex/contrib/test/optimizers/test_distributed_fused_lamb.py
imports: import os
  import inspect
  import torch
  from torch.cuda.amp import GradScaler
  from torch.testing._internal import common_utils
  from apex.parallel.distributed import flat_dist_call
  from apex.contrib.optimizers.distributed_fused_lamb import DistributedFusedLAMB
  from apex.transformer.testing.distributed_test_base import NcclDistributedTestBase
label_desc: Use image node_id 1 to create a new ModelFoo object from inherited base classes: torch.nn with example: obj = ModelFoo()