將寫好的模型進(jìn)行修剪再結(jié)合

創(chuàng)建模型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) )

總結(jié)

以上是生活随笔為你收集整理的nn.Sequential(*modules)的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。