在 PyTorch 中加載 resnet50 prettriated model

Question

我想使用 resnet50 預訓練 model 使用 PyTorch 並且我正在使用以下代碼來加載它：

    import torch
    model = torch.hub.load("pytorch/vision", "resnet50", weights="IMAGENET1K_V2")

雖然我升級了 torchvision 但我收到以下錯誤：

任何想法？

Answer 1

根據最新定義，我們現在使用 torchvision 庫加載模型，您可以嘗試使用：

from torchvision.models import resnet50, ResNet50_Weights

# Old weights with accuracy 76.130%
model1 = resnet50(weights=ResNet50_Weights.IMAGENET1K_V1)

# New weights with accuracy 80.858%
model2 = resnet50(weights=ResNet50_Weights.IMAGENET1K_V2)