捷徑

ShuffleNet V2

ShuffleNet V2 模型基於 ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design 論文。

模型建構器

以下模型建構器可以用於實例化 ShuffleNetV2 模型,無論是否使用預訓練權重。所有模型建構器內部都依賴 torchvision.models.shufflenetv2.ShuffleNetV2 基礎類別。請參考 原始碼 以取得關於此類別的更多詳細資訊。

shufflenet_v2_x0_5(*[, weights, progress])

建構一個具有 0.5 倍輸出通道的 ShuffleNetV2 架構,如 ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design 中所述。

shufflenet_v2_x1_0(*[, weights, progress])

建構一個具有 1.0 倍輸出通道的 ShuffleNetV2 架構,如 ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design 中所述。

shufflenet_v2_x1_5(*[, weights, progress])

建構一個具有 1.5 倍輸出通道的 ShuffleNetV2 架構,如 ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design 中所述。

shufflenet_v2_x2_0(*[, weights, progress])

建構一個具有 2.0 倍輸出通道的 ShuffleNetV2 架構,如 ShuffleNet V2: Practical Guidelines for Efficient CNN Architecture Design 中所述。

文件

存取全面的 PyTorch 開發者文件

查看文件

教學

為初學者和進階開發者取得深入的教學

查看教學

資源

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

查看資源