Функция hub.set_dir

Функция hub.set_dir устанавливает глобальный каталог для загрузки и кэширования моделей, скачанных через PyTorch Hub. Первым параметром функция принимает путь к каталогу в виде строки. По умолчанию используется каталог .cache/torch/hub в домашней директории пользователя. Изменение каталога позволяет контролировать расположение загруженных моделей и управлять дисковым пространством.

Синтаксис

torch.hub.set_dir(dir_path)

Пример

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

import torch

torch.hub.set_dir('./models')
print(torch.hub.get_dir())

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

"./models"

Пример

Давайте установим абсолютный путь к каталогу:

import torch

torch.hub.set_dir('/data/pytorch_models')
print(torch.hub.get_dir())

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

"/data/pytorch_models"

Пример

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

import torch

torch.hub.set_dir('./hub_cache')
model = torch.hub.load('pytorch/vision', 'resnet18', pretrained=False)
print(torch.hub.get_dir())

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

"./hub_cache"

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

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