code: def __init__(self, mode="sq"):
      super(Net, self).__init__()
  
      if mode == "square":
          self.mode = 0
          self.param = torch.nn.Parameter(
              torch.FloatTensor([1.0, -1.0])
          )
      else:
          self.mode = 1
          self.param = torch.nn.Parameter(
              torch.FloatTensor([1.0, -1.0, 1.0])
          )
  
file_path: horovod/examples/adasum/adasum_small_model.py
imports: import torch
  import random
  import horovod.torch
  import numpy
  import argparse
  import os
label_desc: Use image node_id 1 to create a new Net object from inherited base classes: torch.nn with example: obj = Net(mode)