torch.utils.model_zoo

Создано: 09 янв. 2017 | Последнее обновление: 11 июн. 2025

Перенесено в torch.hub.

torch.utils.model_zoo.load_url(url, model_dir=None, map_location=None, progress=True, check_hash=False, file_name=None, weights_only=False) [source]

Загружает сериализованный объект Torch по указанному URL.

Если загруженный файл является zip-файлом, он будет автоматически распакован.

Если объект уже находится в model_dir, он десериализуется и возвращается. Значение по умолчанию для model_dir — <hub_dir>/checkpoints, где hub_dir — это каталог, возвращаемый get_dir().

Параметры:

• url (str) – URL объекта для загрузки
• model_dir (str, необязательно) – каталог для сохранения объекта
• map_location (необязательно) – функция или словарь, указывающие, как переназначить расположение хранилищ (см. torch.load)
• progress (bool, необязательно) – нужно ли отображать индикатор выполнения в stderr. По умолчанию: True
• check_hash (bool, необязательно) – Если True, часть имени файла в URL должна соответствовать соглашению об именовании filename-<sha256>.ext, где <sha256> — первые восемь или более цифр хеша SHA256 содержимого файла. Хеш используется для обеспечения уникальности имен и проверки содержимого файла. По умолчанию: False
• file_name (str, необязательно) – имя загруженного файла. Если параметр не задан, будет использовано имя файла из url.
• weights_only (bool, необязательно) – Если True, будут загружены только веса, без сложных объектов, сериализованных с помощью pickle. Рекомендуется для ненадежных источников. Подробнее см. load().

Тип возвращаемого значения:

dict[str, Any]

Пример

>>> state_dict = torch.hub.load_state_dict_from_url(
...     "https://s3.amazonaws.com/pytorch/models/resnet18-5c106cde.pth"
... )