code: def _make_stages(in_channels, out_channels, bin_sz):
      prior = nn.AdaptiveAvgPool2d(output_size=bin_sz)
      conv = nn.Conv2d(
          in_channels, out_channels, kernel_size=1, bias=False
      )
      bn = nn.BatchNorm2d(out_channels)
      relu = nn.ReLU(inplace=True)
      return nn.Sequential(prior, conv, bn, relu)
  
imports: import torch
  import logging
  import torch.nn
  import torch.nn.functional
  from torchvision import models
  from itertools import chain
  import numpy
  from collections import OrderedDict
label_desc: Use image node_id 2 for calling the PSPModule obj's underlying member method code with example usage: obj._make_stages(in_channels, out_channels, bin_sz) and returns: nn