code: def __init__(self):
      super(NeuralNetwork, self).__init__()
      self.layer1 = nn.Linear(input_size, layer_size)
      self.relu = nn.ReLU()
      self.layer2 = nn.Linear(layer_size, output_size)
  
file_path: ray/doc/source/train/doc_code/hvd_trainer.py
imports: import os
  import tempfile
  import horovod.torch
  import ray
  from ray import train
  from ray.train import Checkpoint, ScalingConfig
  import ray.train.torch
  from ray.train.horovod import HorovodTrainer
  import torch
  import torch.nn
label_desc: Use image node_id 1 to create a new NeuralNetwork object from inherited base classes: nn with example: obj = NeuralNetwork()