code: def __init__(self):
      super(Net, self).__init__()
      self.conv1 = nn.Conv2d(3, 6, 5)
      self.pool = nn.MaxPool2d(2, 2)
      self.conv2 = nn.Conv2d(6, 16, 5)
      self.fc1 = nn.Linear(16 * 5 * 5, 120)
      self.fc2 = nn.Linear(120, 84)
      self.fc3 = nn.Linear(84, 10)
  
imports: from __future__ import print_function
  import argparse
  import matplotlib.pyplot
  import numpy
  import torch
  import torchvision
  import torchvision.transforms
  import torch.nn
  import torch.nn.functional
  import torch.optim
  from zoo.orca import init_orca_context, stop_orca_context
  from zoo.orca.learn.pytorch import Estimator
  from zoo.orca.learn.metrics import Accuracy
  from zoo.orca.learn.trigger import EveryEpoch
label_desc: Use image node_id 1 to create a new Net object from inherited base classes: nn with example: obj = Net()