Как загрузить предварительно обученную модель googlenet в pytorch

Question

Я пытаюсь настроить сеть GoogleNet по определенному набору данных, но у меня возникают проблемы при загрузке.Теперь я пытаюсь:

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

Однако я получаю сообщение об ошибке:

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

У меня установлена ​​последняя версия torchvision, но она была переустановлена, чтобы быть уверенной, что ошибка все еще существует.

Answer 1

Вместо этого вы можете использовать модель GoogLeNet inception_v3 ( "Переосмысление начальной архитектуры для компьютерного зрения" ):

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