torch.utils.model_zoo

移至 torch.hub。

torch.utils.model_zoo。load_url（url， model_dir=无， map_location=无， progress=True， check_hash=False， file_name = 无， weights_only = False)

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

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

如果对象已经存在于 model_dir 中，则会对其进行反序列化，并且 返回。 的默认值 是 where，其中 是 返回的目录。model_dir<hub_dir>/checkpointshub_dir

参数

• url （str） – 要下载的对象的 URL

• model_dir （str， optional） – 保存对象的目录

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

• progress （bool， optional） – 是否向 stderr 显示进度条。 默认值：True

• check_hash （bool， optional） – 如果为 True，则 URL 的文件名部分应遵循命名约定，其中前 8 个或更多 文件内容的 SHA256 哈希值的数字。哈希值用于 确保名称唯一并验证文件的内容。 默认值：Falsefilename-<sha256>.ext<sha256>

• file_name （str， optional） – 下载文件的名称。如果未设置，则将使用 Filename from。url

• weights_only （bool， optional） – 如果为 True，则仅加载权重，而不会加载复杂的腌制对象。 建议用于不受信任的源。有关更多详细信息，请参阅。

返回类型

Dict[str， 任意]

例

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