code: def __init__(self):
      super(TestModel, self).__init__()
      self.fc1 = nn.Linear(4, 256)
      self.fc2 = nn.Linear(256, 128)
      self.fc3 = nn.Linear(128, 1)
  
imports: import numpy
  import unittest
  import paddle
  import paddle.nn
  import parl
  import os
label_desc: Use image node_id 1 to create a new TestModel object from inherited base classes: parl with example: obj = TestModel()