Функция hub.help

Функция hub.help выводит документацию по модели, загруженной из PyTorch Hub. Первым параметром функция принимает имя репозитория, вторым - название модели, для которой требуется справка. Функция полезна для получения информации о входных и выходных параметрах модели, а также о возможных аргументах.

Синтаксис

torch.hub.help(repo, model)

Пример

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

import torch

torch.hub.help('pytorch/vision', 'resnet50')

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

"Help on model resnet50:
Pretrained ResNet-50 model.
Arguments:
	pretrained (bool): If True, returns a model pre-trained on ImageNet.
	progress (bool): If True, displays a progress bar of the download to stderr.
Returns:
	model (torch.nn.Module): ResNet-50 model."

Пример

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

import torch

torch.hub.help('pytorch/vision', 'alexnet')

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

"Help on model alexnet:
AlexNet model architecture from the paper 'ImageNet Classification with Deep Convolutional Neural Networks'.
Arguments:
	pretrained (bool): If True, returns a model pre-trained on ImageNet.
	progress (bool): If True, displays a progress bar of the download to stderr.
Returns:
	model (torch.nn.Module): AlexNet model."

Пример

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

import torch

torch.hub.help('pytorch/vision', 'squeezenet1_0')

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

"Help on model squeezenet1_0:
SqueezeNet model architecture from the paper 'SqueezeNet: AlexNet-level accuracy with 50x fewer parameters and <0.5MB model size'.
Arguments:
	pretrained (bool): If True, returns a model pre-trained on ImageNet.
	progress (bool): If True, displays a progress bar of the download to stderr.
Returns:
	model (torch.nn.Module): SqueezeNet model."

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

функцию hub.load,
которая загружает модель из репозитория PyTorch Hub
функцию hub.list,
которая возвращает список доступных моделей в репозитории
функцию hub.load_state_dict_from_url,
которая загружает состояние модели по URL-адресу
функцию hub.set_dir,
которая задаёт директорию для кэширования загруженных моделей