code: def __init__(self, ch, squeeze_ch):
      super().__init__()
      self.se = nn.Sequential(
          nn.AdaptiveAvgPool2d(1),
          nn.Conv2d(ch, squeeze_ch, 1, 1, 0, bias=True),
          Swish(),
          nn.Conv2d(squeeze_ch, ch, 1, 1, 0, bias=True),
      )
  
file_path: pywick/pywick/models/segmentation/testnets/mixnet/layers.py
imports: import torch
  import torch.nn
label_desc: Use image node_id 1 to create a new SEModule object from inherited base classes: nn with example: obj = SEModule(ch, squeeze_ch)