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

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

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

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

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

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


python pytorch conv-neural-network torchvision pre-trained-model
person ice    schedule 19.04.2019    source источник

Ответы (1)


arrow_upward
0
arrow_downward

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

import torchvision
google_net = torchvision.models.inception_v3(pretrained=True)
person kmario23    schedule 19.04.2019