使用情景:调用torchvision的API,想用在ImageNet数据集上的restnet模型,想用pretrained的时候报的错误。 大致意思是参数列表中的pretrained在新版本中被废用了,要使用weights这个参数。
解决办法:
from torchvision.models import resnet18, ResNet18_Weights
# 使用预训练的 ResNet18 作为特征提取器
self.feature_extractor = resnet18(weights=ResNet18_Weights.IMAGENET1K_V1)
访问该网页,查询对应模型的权重:
https://pytorch.org/vision/main/models.html