code: def __init__(self):
      super().__init__()
      self._bottom_model = None
      self._top_model = None
      self._agg_layer = None
      self._ctx = None
      self.device = None
  
imports: import torch
  import torch
  from fate.arch import Context
  from fate.ml.nn.model_zoo.agg_layer.agg_layer import AggLayerGuest, AggLayerHost
  from fate.ml.nn.model_zoo.agg_layer.fedpass.agg_layer import FedPassAggLayerGuest, FedPassAggLayerHost, get_model
  from fate.ml.nn.model_zoo.agg_layer.sshe.agg_layer import SSHEAggLayerHost, SSHEAggLayerGuest
  from typing import Any, Dict, List, Union, Callable, Literal
  from dataclasses import dataclass, fields
  from enum import Enum
  import logging
  from torch import device
label_desc: Use image node_id 1 to create a new HeteroNNModelBase object from inherited base classes: t.nn with example: obj = HeteroNNModelBase()