code: def __init__(self):
      super(MyModel, self).__init__()
      self.conv1 = nn.Conv2d(3, 6, 3, 1, 1)
      self.bn1 = nn.BatchNorm2d(6)
      self.param = nn.Parameter(torch.randn(1))
  
imports: import unittest
  import torch
  import torch.nn
  import torch.nn.functional
  import torch.optim
  from apex import amp
  from utils import common_init, FLOAT
label_desc: Use image node_id 1 to create a new MyModel object from inherited base classes: torch.nn with example: obj = MyModel()