捷徑

SwinTransformer

SwinTransformer 模型基於 Swin Transformer: Hierarchical Vision Transformer using Shifted Windows 論文。 SwinTransformer V2 模型基於 Swin Transformer V2: Scaling Up Capacity and Resolution 論文。

模型建構器

以下模型建構器可用於實例化具有和不具有預訓練權重的 SwinTransformer 模型(原始和 V2)。 所有模型建構器都在內部依賴 torchvision.models.swin_transformer.SwinTransformer 基底類別。 請參閱 原始程式碼 以取得有關此類別的更多詳細資訊。

swin_t(*[, weights, progress])

Swin Transformer: Hierarchical Vision Transformer using Shifted Windows 建構 swin_tiny 架構。

swin_s(*[, weights, progress])

Swin Transformer: Hierarchical Vision Transformer using Shifted Windows 建構 swin_small 架構。

swin_b(*[, weights, progress])

Swin Transformer: Hierarchical Vision Transformer using Shifted Windows 建構 swin_base 架構。

swin_v2_t(*[, weights, progress])

Swin Transformer V2: Scaling Up Capacity and Resolution 建構 swin_v2_tiny 架構。

swin_v2_s(*[, weights, progress])

Swin Transformer V2: Scaling Up Capacity and Resolution 建構 swin_v2_small 架構。

swin_v2_b(*[, weights, progress])

Swin Transformer V2: Scaling Up Capacity and Resolution 建構 swin_v2_base 架構。

文件

存取 PyTorch 的完整開發人員文件

檢視文件

教學

取得初學者和進階開發人員的深入教學課程

查看教學課程

資源

尋找開發資源並獲得問題解答

查看資源