vgg16的权重更新

我们在获取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”。

猜你喜欢

转载自blog.csdn.net/weixin_67957872/article/details/127173609
今日推荐