code: def __init__(self, mlahead_channels=128):
      super(MLAHeads, self).__init__()
      self.head2 = nn.Sequential(
          layers.ConvBNReLU(
              mlahead_channels * 2,
              mlahead_channels,
              3,
              padding=1,
              bias_attr=False,
          ),
          layers.ConvBNReLU(
              mlahead_channels,
              mlahead_channels,
              3,
              padding=1,
              bias_attr=False,
          ),
      )
      self.head3 = nn.Sequential(
          layers.ConvBNReLU(
              mlahead_channels * 2,
              mlahead_channels,
              3,
              padding=1,
              bias_attr=False,
          ),
          layers.ConvBNReLU(
              mlahead_channels,
              mlahead_channels,
              3,
              padding=1,
              bias_attr=False,
          ),
      )
      self.head4 = nn.Sequential(
          layers.ConvBNReLU(
              mlahead_channels * 2,
              mlahead_channels,
              3,
              padding=1,
              bias_attr=False,
          ),
          layers.ConvBNReLU(
              mlahead_channels,
              mlahead_channels,
              3,
              padding=1,
              bias_attr=False,
          ),
      )
      self.head5 = nn.Sequential(
          layers.ConvBNReLU(
              mlahead_channels * 2,
              mlahead_channels,
              3,
              padding=1,
              bias_attr=False,
          ),
          layers.ConvBNReLU(
              mlahead_channels,
              mlahead_channels,
              3,
              padding=1,
              bias_attr=False,
          ),
      )
  
file_path: PaddleSeg/paddleseg/models/mla_transformer.py
imports: import paddle
  import paddle.nn
  import paddle.nn.functional
  from paddleseg.models import layers
  from paddleseg.cvlibs import manager
  from paddleseg.utils import utils
label_desc: Use image node_id 1 to create a new MLAHeads object from inherited base classes: nn with example: obj = MLAHeads(mlahead_channels)