Функция hub.list

Функция hub.list возвращает список всех доступных моделей в указанном репозитории PyTorch Hub. Первым параметром функция принимает строку с именем репозитория в формате пользователь/репозиторий. Вторым параметром можно передать опциональную ветку или тег репозитория. Функция полезна для предварительного просмотра доступных моделей перед их загрузкой с помощью hub.load.

Синтаксис

torch.hub.list(repo_or_dir, [branch_or_tag])

Пример

Давайте получим список всех моделей из официального репозитория PyTorch Vision:

import torch

models = torch.hub.list('pytorch/vision')
print(models)

Результат выполнения кода:

[
	'resnet18',
	'resnet34',
	'resnet50',
	'resnet101',
	'resnet152',
	'resnext50_32x4d',
	'resnext101_32x8d',
	'wide_resnet50_2',
	'wide_resnet101_2',
	'squeezenet1_0',
	'squeezenet1_1',
	'densenet121',
	'densenet169',
	'densenet201',
	'densenet161',
	'mobilenet_v2',
	'mobilenet_v3_large',
	'mobilenet_v3_small',
	'shufflenet_v2_x0_5',
	'shufflenet_v2_x1_0',
	'shufflenet_v2_x1_5',
	'shufflenet_v2_x2_0',
	'inception_v3',
	'googlenet',
	'alexnet',
	'vgg11',
	'vgg13',
	'vgg16',
	'vgg19',
	'vgg11_bn',
	'vgg13_bn',
	'vgg16_bn',
	'vgg19_bn',
]

Пример

Давайте получим список моделей из репозитория с указанием конкретной ветки:

import torch

models = torch.hub.list('pytorch/vision:main')
print(len(models))

Результат выполнения кода:

34

Пример

Давайте проверим наличие конкретной модели в репозитории перед её загрузкой:

import torch

available_models = torch.hub.list('pytorch/vision')
target_model = 'resnet50'

if target_model in available_models:
	print('Model available')
else:
	print('Model not found')

Результат выполнения кода:

"Model available"

Пример

Давайте получим список моделей из репозитория huggingface/pytorch-transformers для работы с трансформерами:

import torch

models = torch.hub.list('huggingface/pytorch-transformers')
print(models[:5])

Результат выполнения кода:

[
	'bert-base-uncased',
	'bert-large-uncased',
	'bert-base-cased',
	'bert-large-cased',
	'bert-base-multilingual-uncased'
]

Смотрите также

функцию hub.load,
которая загружает модель из PyTorch Hub
функцию hub.help,
которая показывает справку по модели из Hub
функцию hub.set_dir,
которая устанавливает директорию для загрузки моделей
функцию hub.get_dir,
которая возвращает текущую директорию загрузки