vgg16的权重更新

我们在获取vgg16网络模型参数时,如果用如下方式获取会出错。

vgg16_false = torchvision.models.vgg16(pretrained=False)
vgg16_true = torchvision.models.vgg16(pretrained=True)

vgg16的权重更新_第1张图片

 可以将代码更改如下:

vgg16_false = torchvision.models.vgg16(weights=None)

vgg16_true = torchvision.models.vgg16(weights=VGG16_Weights.DEFAULT)

再运行就不会报错了。

错误原因:参数pretrained自0.13起已弃用,将在0.15后删除,要改用“weights”。

你可能感兴趣的:(人工智能,深度学习)