Загрузка из hub в PyTorch
Модель из чужого репозитория GitHub
подтягивают через каталог Hub
в PyTorch. Функция hub.load
принимает имя репозитория и имя
модели из его списка, а на выходе
отдаёт готовый модуль.
Запросим сеть без предобученных
весов из репозитория
pytorch/vision и выведем
тип полученного объекта.
При первом обращении среда
может скачать файлы репозитория.
Если загрузка не удалась,
дело в сети или в версии
пакетов, а не в самой строке
вызова:
import torch
model = torch.hub.load(
'pytorch/vision:v0.10.0',
'resnet18',
pretrained=False,
)
print(type(model))
Загрузите модель alexnet
из репозитория pytorch/vision
без предобученных весов
и выведите тип полученного
объекта.
Загрузите модель squeezenet1_0
из того же репозитория
без предобученных весов
и выведите имя класса
типа объекта.
Загрузите модель mobilenet_v2
из каталога Hub для
pytorch/vision без готовых
весов и выведите, является
ли результат экземпляром
torch.nn.Module.