РЕПЕТИТОР математика физика информатика
Для школьников и студентов. Подтягивание пробелов. ЦЭ, ЦТ, ОГЭ, ЕГЭ.
Идет набор на ЛЕТО. Жмите для подробностей:)
⊗pytoSpHbLd 79 of 82 menu
◀ ▶

Загрузка из 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.

← →
↑
Мы используем cookie для работы сайта, аналитики и персонализации. Обработка данных происходит согласно Политике конфиденциальности.
принять все настроить отклонить