如何在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)