code: def __init__(self, inplanes=64, embed_dim=384, with_cp=False):
      super().__init__()
      self.with_cp = with_cp
  
      self.stem = nn.Sequential(
          *[
              nn.Conv2d(
                  3,
                  inplanes,
                  kernel_size=3,
                  stride=2,
                  padding=1,
                  bias=False,
              ),
              nn.SyncBatchNorm(inplanes),
              nn.ReLU(inplace=True),
              nn.Conv2d(
                  inplanes,
                  inplanes,
                  kernel_size=3,
                  stride=1,
                  padding=1,
                  bias=False,
              ),
              nn.SyncBatchNorm(inplanes),
              nn.ReLU(inplace=True),
              nn.Conv2d(
                  inplanes,
                  inplanes,
                  kernel_size=3,
                  stride=1,
                  padding=1,
                  bias=False,
              ),
              nn.SyncBatchNorm(inplanes),
              nn.ReLU(inplace=True),
              nn.MaxPool2d(kernel_size=3, stride=2, padding=1),
          ]
      )
      self.conv2 = nn.Sequential(
          *[
              nn.Conv2d(
                  inplanes,
                  2 * inplanes,
                  kernel_size=3,
                  stride=2,
                  padding=1,
                  bias=False,
              ),
              nn.SyncBatchNorm(2 * inplanes),
              nn.ReLU(inplace=True),
          ]
      )
      self.conv3 = nn.Sequential(
          *[
              nn.Conv2d(
                  2 * inplanes,
                  4 * inplanes,
                  kernel_size=3,
                  stride=2,
                  padding=1,
                  bias=False,
              ),
              nn.SyncBatchNorm(4 * inplanes),
              nn.ReLU(inplace=True),
          ]
      )
      self.conv4 = nn.Sequential(
          *[
              nn.Conv2d(
                  4 * inplanes,
                  4 * inplanes,
                  kernel_size=3,
                  stride=2,
                  padding=1,
                  bias=False,
              ),
              nn.SyncBatchNorm(4 * inplanes),
              nn.ReLU(inplace=True),
          ]
      )
      self.fc1 = nn.Conv2d(
          inplanes,
          embed_dim,
          kernel_size=1,
          stride=1,
          padding=0,
          bias=True,
      )
      self.fc2 = nn.Conv2d(
          2 * inplanes,
          embed_dim,
          kernel_size=1,
          stride=1,
          padding=0,
          bias=True,
      )
      self.fc3 = nn.Conv2d(
          4 * inplanes,
          embed_dim,
          kernel_size=1,
          stride=1,
          padding=0,
          bias=True,
      )
      self.fc4 = nn.Conv2d(
          4 * inplanes,
          embed_dim,
          kernel_size=1,
          stride=1,
          padding=0,
          bias=True,
      )
  
imports: from functools import partial
  import torch
  import torch.nn
  import torch.utils.checkpoint
  from ...ops.modules import MSDeformAttn
  from .drop_path import DropPath
label_desc: Use image node_id 1 to create a new SpatialPriorModule object from inherited base classes: nn with example: obj = SpatialPriorModule(inplanes, embed_dim, with_cp)