Функция hub.get_dir

Функция hub.get_dir возвращает строку с абсолютным путем к директории, которую PyTorch Hub использует для хранения загруженных моделей и файлов. Эта функция не принимает аргументов и полезна для проверки текущего местоположения кэша, а также для организации работы с загруженными моделями.

Синтаксис

torch.hub.get_dir()

Пример

Получим путь к директории кэша PyTorch Hub:

import torch

dir_path = torch.hub.get_dir()
print(dir_path)

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

"/home/user/.cache/torch/hub"

Пример

Проверим, существует ли директория кэша, и создадим ее при необходимости:

import torch
import os

dir_path = torch.hub.get_dir()
if not os.path.exists(dir_path):
	os.makedirs(dir_path)
	print("Directory created")
else:
	print("Directory exists")

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

"Directory exists"

Пример

Используем полученный путь для сохранения информации о загруженных моделях:

import torch
import os

hub_dir = torch.hub.get_dir()
models_dir = os.path.join(hub_dir, "models")

if os.path.exists(models_dir):
	model_files = os.listdir(models_dir)
	print(f"Found {len(model_files)} model directories")
else:
	print("No models downloaded yet")

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

"No models downloaded yet"

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

функцию hub.set_dir,
которая устанавливает пользовательскую директорию для кэша
функцию hub.load,
которая загружает предобученную модель из репозитория
функцию hub.list,
которая выводит список доступных моделей в репозитории
функцию hub.help,
которая выводит справку по использованию моделей из репозитория