code: def __init__(self):
      super(CNN, self).__init__()
      self.conv1 = nn.Conv2d(1, 16, kernel_size=5, padding=1)
      self.fc1 = nn.Linear(16 * 13 * 13, 100)
      self.fc2 = nn.Linear(100, 2)
  
imports: import unittest
  import crypten
  import crypten.communicator
  import numpy
  import torch
  import torch.distributed
  import torch.nn
  import torch.nn.functional
  from crypten.common import serial
  from crypten.config import cfg
  from test.multiprocess_test_case import get_random_test_tensor, MultiProcessTestCase
label_desc: Use image node_id 1 to create a new CNN object from inherited base classes: nn with example: obj = CNN()