如何在pytorch中加载预训练的googlenet模型

Question

我正在尝试通过特定的数据集微调GoogleNet网络，但无法加载它。我现在尝试的是：

model = torchvision.models.googlenet(pretrained=True)

但是我得到一个错误：

AttributeError: module 'torchvision.models' has no attribute 'googlenet'

我拥有的是Torchvision的最新版本，但是为了确保没有错误，重新安装了它。

Answer 1

您可以改为使用GoogLeNet inception_v3模型（"Rethinking the Inception Architecture for Computer Vision"）：

import torchvision
google_net = torchvision.models.inception_v3(pretrained=True)