code: def __init__(self, num_classes=10):
      super(MobileNetV2, self).__init__()
      # NOTE: change conv1 stride 2 -> 1 for CIFAR10
      self.conv1 = nn.Conv2d(
          3, 32, kernel_size=3, stride=1, padding=1, bias=False
      )
      self.bn1 = nn.BatchNorm2d(32)
      self.layers = self._make_layers(in_planes=32)
      self.conv2 = nn.Conv2d(
          320, 1280, kernel_size=1, stride=1, padding=0, bias=False
      )
      self.bn2 = nn.BatchNorm2d(1280)
      self.linear = nn.Linear(1280, num_classes)
  
file_path: nni/examples/trials/cifar10_pytorch/models/mobilenetv2.py
imports: import torch
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 1 to create a new MobileNetV2 object from inherited base classes: nn with example: obj = MobileNetV2(num_classes)