code: def __init__(self, features, out_features=1024, sizes=(1, 2, 3, 6)):
      super().__init__()
      self.stages = []
      self.stages = nn.ModuleList(
          [self._make_stage(features, size) for size in sizes]
      )
      self.bottleneck = nn.Conv2d(
          features * (len(sizes) + 1), out_features, kernel_size=1
      )
      self.relu = nn.ReLU()
  
file_path: pywick/pywick/models/segmentation/lexpsp.py
imports: from .lex_extractors import *
label_desc: Use image node_id 1 to create a new PSPModule object from inherited base classes: nn with example: obj = PSPModule(features, out_features, sizes)