code: def __init__(self, in_channels, sizes=(1, 2, 3, 6), **kwargs):
      super(_PSPModule, self).__init__()
      out_channels = int(in_channels / 4)
      self.avgpools = nn.ModuleList()
      self.convs = nn.ModuleList()
      for size in sizes:
          self.avgpool.append(nn.AdaptiveAvgPool2d(size))
          self.convs.append(
              _ConvBNReLU(in_channels, out_channels, 1, **kwargs)
          )
  
imports: import torch
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 1 to create a new _PSPModule object from inherited base classes: nn with example: obj = _PSPModule(in_channels, sizes)