code: def __init__(self):
      super(ImageClassifier, self).__init__(BasicBlock, [2, 2, 2, 2])
      self.fc = nn.Sequential(
          nn.Linear(self.fc.in_features, 512),
          nn.ReLU(),
          nn.Dropout(),
          nn.Linear(512, 2),
      )
  
imports: from torchvision.models.resnet import ResNet, BasicBlock
  import torch.nn
label_desc: Use image node_id 1 to create a new ImageClassifier object from inherited base classes: ResNet with example: obj = ImageClassifier()