【踩坑日记29】UserWarning: Arguments other than a weight enum or `None` for ‘weights‘ are deprecated since

问题描述

UserWarning: Arguments other than a weight enum or `None` for 'weights' are deprecated since 0.13 and may be removed in the future. The current behavior is equivalent to passing `weights=VGG16_Weights.IMAGENET1K_V1`. You can also use `weights=VGG16_Weights.DEFAULT` to get the most up-to-date weights.

问题分析

from torchvision import models
model_old = models.vgg16(pretrained=True)

原始代码中pretrained=True已经弃用，要使用weights=VGG16_Weights.IMAGENET1K_V1（指定权重版本）或者weights=VGG16_Weights.DEFAULT（权重选择最新版本）

问题解决

代码改为：

from torchvision import models
model_old = models.vgg16(weights=models.VGG16_Weights.IMAGENET1K_V1)

参考

UserWarning: Arguments other than a weight enum or None for ‘weights‘ are deprecated since 0.13 an
【PyTorch教程】04-详解torchvision 0.13中的预训练模型加载的更新及报错的解决方法 (2022年最新)