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, weights_only=False)

Загружает сериализованный объект 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, будут загружены только веса, и не будут загружаются сложные сериализованные объекты. Рекомендуется для ненадежных источников. См. load() для получения дополнительных сведений.

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

Dict[str, Any]

Пример

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