EfficientPS(
  (backbone): GenEfficientNet(
    (conv_stem): Conv2dSame(3, 48, kernel_size=(3, 3), stride=(2, 2), bias=False)
    (bn1): InPlaceABN(48, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
    (act1): Identity()
    (blocks): Sequential(
      (0): Sequential(
        (0): DepthwiseSeparableConv(
          (conv_dw): Conv2d(48, 48, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=48, bias=False)
          (bn1): InPlaceABN(48, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act1): Identity()
          (se): Identity()
          (conv_pw): Conv2d(48, 24, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn2): InPlaceABN(24, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
        )
        (1): DepthwiseSeparableConv(
          (conv_dw): Conv2d(24, 24, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=24, bias=False)
          (bn1): InPlaceABN(24, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act1): Identity()
          (se): Identity()
          (conv_pw): Conv2d(24, 24, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn2): InPlaceABN(24, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
        )
        (2): DepthwiseSeparableConv(
          (conv_dw): Conv2d(24, 24, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=24, bias=False)
          (bn1): InPlaceABN(24, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act1): Identity()
          (se): Identity()
          (conv_pw): Conv2d(24, 24, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn2): InPlaceABN(24, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
        )
      )
      (1): Sequential(
        (0): InvertedResidual(
          (conv_pw): Conv2d(24, 144, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(144, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2dSame(144, 144, kernel_size=(3, 3), stride=(2, 2), groups=144, bias=False)
          (bn2): InPlaceABN(144, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(144, 40, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(40, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (1): InvertedResidual(
          (conv_pw): Conv2d(40, 240, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(240, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(240, 240, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=240, bias=False)
          (bn2): InPlaceABN(240, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(240, 40, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(40, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (2): InvertedResidual(
          (conv_pw): Conv2d(40, 240, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(240, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(240, 240, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=240, bias=False)
          (bn2): InPlaceABN(240, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(240, 40, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(40, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (3): InvertedResidual(
          (conv_pw): Conv2d(40, 240, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(240, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(240, 240, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=240, bias=False)
          (bn2): InPlaceABN(240, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(240, 40, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(40, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (4): InvertedResidual(
          (conv_pw): Conv2d(40, 240, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(240, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(240, 240, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=240, bias=False)
          (bn2): InPlaceABN(240, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(240, 40, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(40, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
      )
      (2): Sequential(
        (0): InvertedResidual(
          (conv_pw): Conv2d(40, 240, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(240, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2dSame(240, 240, kernel_size=(5, 5), stride=(2, 2), groups=240, bias=False)
          (bn2): InPlaceABN(240, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(240, 64, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(64, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (1): InvertedResidual(
          (conv_pw): Conv2d(64, 384, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(384, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(384, 384, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=384, bias=False)
          (bn2): InPlaceABN(384, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(384, 64, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(64, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (2): InvertedResidual(
          (conv_pw): Conv2d(64, 384, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(384, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(384, 384, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=384, bias=False)
          (bn2): InPlaceABN(384, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(384, 64, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(64, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (3): InvertedResidual(
          (conv_pw): Conv2d(64, 384, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(384, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(384, 384, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=384, bias=False)
          (bn2): InPlaceABN(384, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(384, 64, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(64, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (4): InvertedResidual(
          (conv_pw): Conv2d(64, 384, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(384, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(384, 384, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=384, bias=False)
          (bn2): InPlaceABN(384, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(384, 64, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(64, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
      )
      (3): Sequential(
        (0): InvertedResidual(
          (conv_pw): Conv2d(64, 384, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(384, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2dSame(384, 384, kernel_size=(3, 3), stride=(2, 2), groups=384, bias=False)
          (bn2): InPlaceABN(384, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(384, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(128, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (1): InvertedResidual(
          (conv_pw): Conv2d(128, 768, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(768, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(768, 768, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=768, bias=False)
          (bn2): InPlaceABN(768, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(768, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(128, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (2): InvertedResidual(
          (conv_pw): Conv2d(128, 768, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(768, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(768, 768, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=768, bias=False)
          (bn2): InPlaceABN(768, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(768, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(128, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (3): InvertedResidual(
          (conv_pw): Conv2d(128, 768, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(768, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(768, 768, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=768, bias=False)
          (bn2): InPlaceABN(768, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(768, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(128, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (4): InvertedResidual(
          (conv_pw): Conv2d(128, 768, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(768, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(768, 768, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=768, bias=False)
          (bn2): InPlaceABN(768, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(768, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(128, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (5): InvertedResidual(
          (conv_pw): Conv2d(128, 768, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(768, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(768, 768, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=768, bias=False)
          (bn2): InPlaceABN(768, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(768, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(128, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (6): InvertedResidual(
          (conv_pw): Conv2d(128, 768, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(768, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(768, 768, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=768, bias=False)
          (bn2): InPlaceABN(768, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(768, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(128, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
      )
      (4): Sequential(
        (0): InvertedResidual(
          (conv_pw): Conv2d(128, 768, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(768, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(768, 768, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=768, bias=False)
          (bn2): InPlaceABN(768, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(768, 176, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(176, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (1): InvertedResidual(
          (conv_pw): Conv2d(176, 1056, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(1056, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(1056, 1056, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=1056, bias=False)
          (bn2): InPlaceABN(1056, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(1056, 176, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(176, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (2): InvertedResidual(
          (conv_pw): Conv2d(176, 1056, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(1056, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(1056, 1056, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=1056, bias=False)
          (bn2): InPlaceABN(1056, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(1056, 176, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(176, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (3): InvertedResidual(
          (conv_pw): Conv2d(176, 1056, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(1056, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(1056, 1056, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=1056, bias=False)
          (bn2): InPlaceABN(1056, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(1056, 176, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(176, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (4): InvertedResidual(
          (conv_pw): Conv2d(176, 1056, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(1056, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(1056, 1056, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=1056, bias=False)
          (bn2): InPlaceABN(1056, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(1056, 176, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(176, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (5): InvertedResidual(
          (conv_pw): Conv2d(176, 1056, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(1056, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(1056, 1056, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=1056, bias=False)
          (bn2): InPlaceABN(1056, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(1056, 176, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(176, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (6): InvertedResidual(
          (conv_pw): Conv2d(176, 1056, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(1056, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(1056, 1056, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=1056, bias=False)
          (bn2): InPlaceABN(1056, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(1056, 176, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(176, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
      )
      (5): Sequential(
        (0): InvertedResidual(
          (conv_pw): Conv2d(176, 1056, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(1056, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2dSame(1056, 1056, kernel_size=(5, 5), stride=(2, 2), groups=1056, bias=False)
          (bn2): InPlaceABN(1056, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(1056, 304, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(304, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (1): InvertedResidual(
          (conv_pw): Conv2d(304, 1824, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(1824, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(1824, 1824, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=1824, bias=False)
          (bn2): InPlaceABN(1824, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(1824, 304, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(304, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (2): InvertedResidual(
          (conv_pw): Conv2d(304, 1824, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(1824, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(1824, 1824, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=1824, bias=False)
          (bn2): InPlaceABN(1824, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(1824, 304, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(304, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (3): InvertedResidual(
          (conv_pw): Conv2d(304, 1824, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(1824, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(1824, 1824, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=1824, bias=False)
          (bn2): InPlaceABN(1824, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(1824, 304, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(304, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (4): InvertedResidual(
          (conv_pw): Conv2d(304, 1824, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(1824, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(1824, 1824, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=1824, bias=False)
          (bn2): InPlaceABN(1824, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(1824, 304, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(304, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (5): InvertedResidual(
          (conv_pw): Conv2d(304, 1824, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(1824, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(1824, 1824, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=1824, bias=False)
          (bn2): InPlaceABN(1824, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(1824, 304, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(304, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (6): InvertedResidual(
          (conv_pw): Conv2d(304, 1824, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(1824, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(1824, 1824, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=1824, bias=False)
          (bn2): InPlaceABN(1824, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(1824, 304, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(304, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (7): InvertedResidual(
          (conv_pw): Conv2d(304, 1824, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(1824, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(1824, 1824, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=1824, bias=False)
          (bn2): InPlaceABN(1824, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(1824, 304, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(304, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (8): InvertedResidual(
          (conv_pw): Conv2d(304, 1824, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(1824, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(1824, 1824, kernel_size=(5, 5), stride=(1, 1), padding=(2, 2), groups=1824, bias=False)
          (bn2): InPlaceABN(1824, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(1824, 304, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(304, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
      )
      (6): Sequential(
        (0): InvertedResidual(
          (conv_pw): Conv2d(304, 1824, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(1824, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(1824, 1824, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=1824, bias=False)
          (bn2): InPlaceABN(1824, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(1824, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(512, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (1): InvertedResidual(
          (conv_pw): Conv2d(512, 3072, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(3072, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(3072, 3072, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=3072, bias=False)
          (bn2): InPlaceABN(3072, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(3072, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(512, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
        (2): InvertedResidual(
          (conv_pw): Conv2d(512, 3072, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn1): InPlaceABN(3072, eps=0.001, momentum=0.1, affine=True, activation=identity)
          (act1): LeakyReLU(negative_slope=0.01)
          (conv_dw): Conv2d(3072, 3072, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=3072, bias=False)
          (bn2): InPlaceABN(3072, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
          (act2): Identity()
          (se): Identity()
          (conv_pwl): Conv2d(3072, 512, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn3): InPlaceABN(512, eps=0.001, momentum=0.1, affine=True, activation=identity)
        )
      )
    )
    (conv_head): Conv2d(512, 2048, kernel_size=(1, 1), stride=(1, 1), bias=False)
    (bn2): InPlaceABN(2048, eps=0.001, momentum=0.1, affine=True, activation=leaky_relu[0.01])
    (act2): Identity()
  )
  (neck): TWOWAYFPN(
    (lateral_convs_top): ModuleList(
      (0): ConvModule(
        (conv): Conv2d(40, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
        (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
      )
      (1): ConvModule(
        (conv): Conv2d(64, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
        (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
      )
      (2): ConvModule(
        (conv): Conv2d(176, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
        (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
      )
      (3): ConvModule(
        (conv): Conv2d(2048, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
        (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
      )
    )
    (lateral_convs_bottom): ModuleList(
      (0): ConvModule(
        (conv): Conv2d(2048, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
        (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
      )
      (1): ConvModule(
        (conv): Conv2d(176, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
        (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
      )
      (2): ConvModule(
        (conv): Conv2d(64, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
        (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
      )
      (3): ConvModule(
        (conv): Conv2d(40, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
        (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
      )
    )
    (fpn_convs): ModuleList(
      (0): ConvModule(
        (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
        (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
      )
      (1): ConvModule(
        (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
        (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
      )
      (2): ConvModule(
        (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
        (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
      )
      (3): ConvModule(
        (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), bias=False)
        (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
      )
    )
  )
  (rpn_head): RPNHead(
    (loss_cls): CrossEntropyLoss()
    (loss_bbox): SmoothL1Loss()
    (rpn_conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1))
    (rpn_cls): Conv2d(256, 3, kernel_size=(1, 1), stride=(1, 1))
    (rpn_reg): Conv2d(256, 12, kernel_size=(1, 1), stride=(1, 1))
  )
  (bbox_roi_extractor): SingleRoIExtractor(
    (roi_layers): ModuleList(
      (0): RoIAlign(out_size=(7, 7), spatial_scale=0.25, sample_num=2, use_torchvision=False, aligned=False)
      (1): RoIAlign(out_size=(7, 7), spatial_scale=0.125, sample_num=2, use_torchvision=False, aligned=False)
      (2): RoIAlign(out_size=(7, 7), spatial_scale=0.0625, sample_num=2, use_torchvision=False, aligned=False)
      (3): RoIAlign(out_size=(7, 7), spatial_scale=0.03125, sample_num=2, use_torchvision=False, aligned=False)
    )
  )
  (bbox_head): SharedFCBBoxHead(
    (loss_cls): CrossEntropyLoss()
    (loss_bbox): SmoothL1Loss()
    (fc_cls): Linear(in_features=1024, out_features=9, bias=True)
    (fc_reg): Linear(in_features=1024, out_features=36, bias=True)
    (shared_convs): ModuleList()
    (shared_fcs): ModuleList(
      (0): Linear(in_features=12544, out_features=1024, bias=True)
      (1): Linear(in_features=1024, out_features=1024, bias=True)
    )
    (cls_convs): ModuleList()
    (cls_fcs): ModuleList()
    (reg_convs): ModuleList()
    (reg_fcs): ModuleList()
    (relu): ReLU(inplace=True)
  )
  (mask_roi_extractor): SingleRoIExtractor(
    (roi_layers): ModuleList(
      (0): RoIAlign(out_size=(14, 14), spatial_scale=0.25, sample_num=2, use_torchvision=False, aligned=False)
      (1): RoIAlign(out_size=(14, 14), spatial_scale=0.125, sample_num=2, use_torchvision=False, aligned=False)
      (2): RoIAlign(out_size=(14, 14), spatial_scale=0.0625, sample_num=2, use_torchvision=False, aligned=False)
      (3): RoIAlign(out_size=(14, 14), spatial_scale=0.03125, sample_num=2, use_torchvision=False, aligned=False)
    )
  )
  (mask_head): FCNSepMaskHead(
    (loss_mask): CrossEntropyLoss()
    (convs): ModuleList(
      (0): DepthwiseSeparableConvModule(
        (depthwise_conv): ConvModule(
          (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=256, bias=False)
          (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
        )
        (pointwise_conv): ConvModule(
          (conv): Conv2d(256, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
        )
      )
      (1): DepthwiseSeparableConvModule(
        (depthwise_conv): ConvModule(
          (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=256, bias=False)
          (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
        )
        (pointwise_conv): ConvModule(
          (conv): Conv2d(256, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
        )
      )
      (2): DepthwiseSeparableConvModule(
        (depthwise_conv): ConvModule(
          (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=256, bias=False)
          (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
        )
        (pointwise_conv): ConvModule(
          (conv): Conv2d(256, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
        )
      )
      (3): DepthwiseSeparableConvModule(
        (depthwise_conv): ConvModule(
          (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=256, bias=False)
          (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
        )
        (pointwise_conv): ConvModule(
          (conv): Conv2d(256, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
        )
      )
    )
    (upsample): ConvTranspose2d(256, 256, kernel_size=(2, 2), stride=(2, 2))
    (conv_logits): Conv2d(256, 9, kernel_size=(1, 1), stride=(1, 1))
    (relu): ReLU(inplace=True)
  )
  (semantic_head): EfficientPSSemanticHead(
    (lateral_convs_ss): ModuleList(
      (0): DPC(
        (convs): ModuleList(
          (0): DepthwiseSeparableConvModule(
            (depthwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 6), dilation=(1, 6), groups=256, bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
            (pointwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
          )
          (1): DepthwiseSeparableConvModule(
            (depthwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=256, bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
            (pointwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
          )
          (2): DepthwiseSeparableConvModule(
            (depthwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(6, 21), dilation=(6, 21), groups=256, bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
            (pointwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
          )
          (3): DepthwiseSeparableConvModule(
            (depthwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(18, 15), dilation=(18, 15), groups=256, bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
            (pointwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
          )
          (4): DepthwiseSeparableConvModule(
            (depthwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(6, 3), dilation=(6, 3), groups=256, bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
            (pointwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
          )
        )
        (conv): ConvModule(
          (conv): Conv2d(1280, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn): InPlaceABN(128, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
        )
      )
      (1): DPC(
        (convs): ModuleList(
          (0): DepthwiseSeparableConvModule(
            (depthwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 6), dilation=(1, 6), groups=256, bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
            (pointwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
          )
          (1): DepthwiseSeparableConvModule(
            (depthwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=256, bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
            (pointwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
          )
          (2): DepthwiseSeparableConvModule(
            (depthwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(6, 21), dilation=(6, 21), groups=256, bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
            (pointwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
          )
          (3): DepthwiseSeparableConvModule(
            (depthwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(18, 15), dilation=(18, 15), groups=256, bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
            (pointwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
          )
          (4): DepthwiseSeparableConvModule(
            (depthwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(6, 3), dilation=(6, 3), groups=256, bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
            (pointwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(1, 1), stride=(1, 1), bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
          )
        )
        (conv): ConvModule(
          (conv): Conv2d(1280, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)
          (bn): InPlaceABN(128, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
        )
      )
    )
    (lateral_convs_ls): ModuleList(
      (0): LSFE(
        (convs): Sequential(
          (0): DepthwiseSeparableConvModule(
            (depthwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=256, bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
            (pointwise_conv): ConvModule(
              (conv): Conv2d(256, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)
              (bn): InPlaceABN(128, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
          )
          (1): DepthwiseSeparableConvModule(
            (depthwise_conv): ConvModule(
              (conv): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=128, bias=False)
              (bn): InPlaceABN(128, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
            (pointwise_conv): ConvModule(
              (conv): Conv2d(128, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)
              (bn): InPlaceABN(128, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
          )
        )
      )
      (1): LSFE(
        (convs): Sequential(
          (0): DepthwiseSeparableConvModule(
            (depthwise_conv): ConvModule(
              (conv): Conv2d(256, 256, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=256, bias=False)
              (bn): InPlaceABN(256, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
            (pointwise_conv): ConvModule(
              (conv): Conv2d(256, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)
              (bn): InPlaceABN(128, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
          )
          (1): DepthwiseSeparableConvModule(
            (depthwise_conv): ConvModule(
              (conv): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=128, bias=False)
              (bn): InPlaceABN(128, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
            (pointwise_conv): ConvModule(
              (conv): Conv2d(128, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)
              (bn): InPlaceABN(128, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
          )
        )
      )
    )
    (aligning_convs): ModuleList(
      (0): MC(
        (convs): Sequential(
          (0): DepthwiseSeparableConvModule(
            (depthwise_conv): ConvModule(
              (conv): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=128, bias=False)
              (bn): InPlaceABN(128, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
            (pointwise_conv): ConvModule(
              (conv): Conv2d(128, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)
              (bn): InPlaceABN(128, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
          )
          (1): DepthwiseSeparableConvModule(
            (depthwise_conv): ConvModule(
              (conv): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=128, bias=False)
              (bn): InPlaceABN(128, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
            (pointwise_conv): ConvModule(
              (conv): Conv2d(128, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)
              (bn): InPlaceABN(128, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
          )
        )
      )
      (1): MC(
        (convs): Sequential(
          (0): DepthwiseSeparableConvModule(
            (depthwise_conv): ConvModule(
              (conv): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=128, bias=False)
              (bn): InPlaceABN(128, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
            (pointwise_conv): ConvModule(
              (conv): Conv2d(128, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)
              (bn): InPlaceABN(128, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
          )
          (1): DepthwiseSeparableConvModule(
            (depthwise_conv): ConvModule(
              (conv): Conv2d(128, 128, kernel_size=(3, 3), stride=(1, 1), padding=(1, 1), groups=128, bias=False)
              (bn): InPlaceABN(128, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
            (pointwise_conv): ConvModule(
              (conv): Conv2d(128, 128, kernel_size=(1, 1), stride=(1, 1), bias=False)
              (bn): InPlaceABN(128, eps=1e-05, momentum=0.1, affine=True, activation=leaky_relu[0.01])
            )
          )
        )
      )
    )
    (conv_logits): Conv2d(512, 19, kernel_size=(1, 1), stride=(1, 1))
    (criterion): CrossEntropyLoss()
  )
)