我们在获取vgg16网络模型参数时,如果用如下方式获取会出错。
vgg16_false = torchvision.models.vgg16(pretrained=False)
vgg16_true = torchvision.models.vgg16(pretrained=True)
可以将代码更改如下:
vgg16_false = torchvision.models.vgg16(weights=None)
vgg16_true = torchvision.models.vgg16(weights=VGG16_Weights.DEFAULT)
再运行就不会报错了。
错误原因:参数pretrained自0.13起已弃用,将在0.15后删除,要改用“weights”。