code: def __init__(self, out_feat=10):
      super(LeNet_Top, self).__init__()
      self.pool = nn.MaxPool2d(2, 2)
      self.fc1 = nn.Linear(16 * 4 * 4, 120)
      self.fc1act = nn.ReLU(inplace=True)
      self.fc2 = nn.Linear(120, 84)
      self.fc2act = nn.ReLU(inplace=True)
      self.fc3 = nn.Linear(84, out_feat)
  
imports: from torch.nn import init
  import sys
  import torch
  import torch
  import torchvision
  from torch import nn
  from datetime import datetime
  from fate.ml.nn.model_zoo.agg_layer.fedpass._passport_block import ConvPassportBlock
  from fate.ml.nn.model_zoo.hetero_nn_model import HeteroNNModelGuest, HeteroNNModelHost
  from fate.ml.nn.hetero.hetero_nn import HeteroNNTrainerGuest, HeteroNNTrainerHost, TrainingArguments, FedPassArgument
label_desc: Use image node_id 1 to create a new LeNet_Top object from inherited base classes: nn with example: obj = LeNet_Top(out_feat)