Verwendungsszenario: Aufrufen der Torchvision-API, Versuch, das Restnet-Modell für den ImageNet-Datensatz zu verwenden, und Versuch, vorab trainierte Fehler zu verwenden. Die allgemeine Bedeutung besteht darin, dass die in der Parameterliste vorab trainierten Parameter in der neuen Version veraltet sind und der Gewichtungsparameter verwendet werden muss.
Lösung:
from torchvision.models import resnet18, ResNet18_Weights
# 使用预训练的 ResNet18 作为特征提取器
self.feature_extractor = resnet18(weights=ResNet18_Weights.IMAGENET1K_V1)
Besuchen Sie diese Webseite, um das Gewicht des entsprechenden Modells abzufragen:
https://pytorch.org/vision/main/models.html