UserWarning: The parameter ‘pretrained‘ is deprecated since 0.13 and will be removed in 0.15, please

使用情景:调用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
UserWarning: The parameter ‘pretrained‘ is deprecated since 0.13 and will be removed in 0.15, please_第1张图片

你可能感兴趣的:(python,pytorch)