code: def __init__(self, in_channels, out_channels, sizes=4):
      super().__init__()
      self.stages = nn.LayerList(
          [
              nn.Conv2D(
                  in_channels,
                  in_channels,
                  kernel_size=3,
                  stride=1,
                  groups=in_channels,
                  padding="same",
                  bias_attr=False,
              )
              for _ in range(sizes)
          ]
      )
      self.project = layers.ConvBNPReLU(
          in_channels * (sizes + 1),
          out_channels,
          1,
          stride=1,
          bias_attr=False,
      )
  
imports: import math
  import paddle
  import paddle.nn
  import paddle.nn.functional
  from paddleseg import utils
  from paddleseg.cvlibs import manager, param_init
  from paddleseg.models import layers
label_desc: Use image node_id 1 to create a new PSPModule object from inherited base classes: nn with example: obj = PSPModule(in_channels, out_channels, sizes)