---

在Kaggle的评论区看到有选手想用Swin Transformer来做回归任务，从这个片段可以窥见修改模型的一些思路，于是记录下来学习一下

---

基于 torchvision Swin_B 模型的特征提取

from torchvision.models.swin_transformer import swin_b, Swin_B_Weights, swin_v2_b, Swin_V2_B_Weights
import torch
import torch.nn as nn