code: def __init__(self, in_channels, num_codes):
      super().__init__()
      self.encoding_project = layers.ConvBNReLU(
          in_channels,
          in_channels,
          1,
      )
      self.encoding = nn.Sequential(
          Encoding(channels=in_channels, num_codes=num_codes),
          nn.BatchNorm1D(num_codes),
          nn.ReLU(),
      )
      self.fc = nn.Sequential(
          nn.Linear(in_channels, in_channels),
          nn.Sigmoid(),
      )
  
imports: import paddle
  import paddle.nn
  import paddle.nn.functional
  from paddleseg.cvlibs import manager
  from paddleseg.models import layers
  from paddleseg.utils import utils
label_desc: Use image node_id 1 to create a new EncModule object from inherited base classes: nn with example: obj = EncModule(in_channels, num_codes)