torch.utils.model_zoo

Перемещено в torch.hub.

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

Загружает сериализованный объект 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 будет использоваться, если не задано.

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

Dict[str, Any]

Пример

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