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)

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

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

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

Parameters

• url (str) – 要下载的对象的URL

• model_dir (str, 可选) – 保存对象的目录

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

• 进度 (布尔值, 可选) – 是否在标准错误输出中显示进度条。 默认值: True

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

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

• weights_only (布尔值, 可选) – 如果为True，则只加载权重，不加载复杂的序列化对象。 推荐用于不可信的来源。更多详情请参见load()。

Return type

字典[字符串, 任意]

示例

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