code: def __init__(self):
      super(TestModel, self).__init__()
      self.fc1 = nn.Linear(1, 1)
      self.conv1 = nn.Conv1d(1, 1, 1)
      self.relu = nn.ReLU()
      self.layer1 = TestModule()
      self.layer2 = TestModule2()
  
imports: from unittest import TestCase
  from mock import patch, Mock
  from torch import nn
  from torchbearer.callbacks.manifold_mixup import ManifoldMixup
  import torchbearer
  import torch
label_desc: Use image node_id 1 to create a new TestModel object from inherited base classes: nn with example: obj = TestModel()