code: def __init__(self):
      super().__init__()
      self.net1 = nn.Linear(10, 10)
      self.relu = nn.ReLU()
      self.net2 = nn.Linear(10, 5)
  
file_path: wandb/tests/functional_tests/t0_main/torch/t3_ddp_basic.py
imports: import os
  import torch
  import torch.distributed
  import torch.multiprocessing
  import torch.nn
  import torch.optim
  import wandb
  from torch.nn.parallel import DistributedDataParallel
label_desc: Use image node_id 1 to create a new ToyModel object from inherited base classes: nn with example: obj = ToyModel()