nn.Sequential(*modules)

将写好的模型进行修剪再结合

创建模型model = 巴拉巴拉
modules = list(model.children())# 可以添加索引[] 用于切片
swintransformer = nn.Sequential(*modules)
print(swintransformer)

Sequential(
  (0): PatchEmbed(
    (proj): Conv2d(3, 96, kernel_size=(4, 4), stride=(4, 4))
    (norm): LayerNorm((96,), eps=1e-05, elementwise_affine=True)
  )
  (1): Dropout(p=0.0, inplace=False)
  (2): ModuleList(
    (0): BasicLayer(
      巴拉巴拉。。。
  (3): LayerNorm((768,), eps=1e-05, elementwise_affine=True)
  (4): AdaptiveAvgPool1d(output_size=1)
  (5): Linear(in_features=768, out_features=5, bias=True)
)