SwinTransformer¶
SwinTransformer 模型基於 Swin Transformer:基於移位視窗的分層 Vision Transformer 論文。SwinTransformer V2 模型基於 Swin Transformer V2:擴大容量和解析度 論文。
模型構建器¶
以下模型構建器可用於例項化帶有或不帶有預訓練權重的 SwinTransformer 模型(原始版和 V2 版)。所有模型構建器內部都依賴於 torchvision.models.swin_transformer.SwinTransformer 基類。請參考 原始碼,瞭解有關此類的更多詳細資訊。
|
根據 Swin Transformer:基於移位視窗的分層 Vision Transformer 構建 swin_tiny 架構。 |
|
根據 Swin Transformer:基於移位視窗的分層 Vision Transformer 構建 swin_small 架構。 |
|
根據 Swin Transformer:基於移位視窗的分層 Vision Transformer 構建 swin_base 架構。 |
|
根據 Swin Transformer V2:擴大容量和解析度 構建 swin_v2_tiny 架構。 |
|
根據 Swin Transformer V2:擴大容量和解析度 構建 swin_v2_small 架構。 |
|
根據 Swin Transformer V2:擴大容量和解析度 構建 swin_v2_base 架構。 |