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: ray/python/ray/train/examples/horovod/horovod_tune_example.py
imports: import time
  import numpy
  import torch
  import ray
  import ray.train.torch
  from ray import train, tune
  from ray.train import ScalingConfig
  from ray.train.horovod import HorovodTrainer
  from ray.tune.tune_config import TuneConfig
  from ray.tune.tuner import Tuner
label_desc: Use image node_id 1 to create a new Net object from inherited base classes: torch.nn with example: obj = Net(mode)