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)[source]†

从给定的 URL 加载 Torch 序列化对象。

如果下载的文件是 zip 文件，它将被自动解压缩。

如果对象已存在于 model_dir 中，它将被反序列化并返回。默认值 model_dir 是 <hub_dir>/checkpoints ，其中 hub_dir 是 get_dir() 返回的目录。

参数:

• 对象下载的 URL（字符串）

• 保存对象的目录（字符串，可选）

• map_location（可选）- 一个函数或字典，指定如何重映射存储位置（参见 torch.load）

• progress（布尔值，可选）- 是否在 stderr 上显示进度条。默认：True

• check_hash (bool, 可选) – 如果为 True，URL 的文件名部分应遵循命名约定 filename-<sha256>.ext ，其中 <sha256> 是文件内容的 SHA256 哈希的前八个或更多数字。该哈希用于确保名称唯一并验证文件内容。默认：False

• file_name (str, 可选) – 下载文件的名称。如果没有设置，将使用 url 的文件名。

• weights_only (bool, 可选) – 如果为 True，则只加载权重，不加载复杂的 pickled 对象。建议用于不可信的来源。见 load() 了解更多详情。

返回类型:

dict[str, Any]

示例

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