doc22_092.md

# torch.utils.model_zoo

> [`pytorch.org/docs/stable/model_zoo.html`](https://pytorch.org/docs/stable/model_zoo.html)

移至 torch.hub。

```py
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`的默认值是`<hub_dir>/checkpoints`，其中`hub_dir`是由`get_dir()`返回的目录。

参数

+   **url**（[*str*](https://docs.python.org/3/library/stdtypes.html#str "(in Python v3.12)")） - 要下载的对象的 URL

+   **model_dir**（[*str*](https://docs.python.org/3/library/stdtypes.html#str "(in Python v3.12)")*，*可选） - 保存对象的目录

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

+   **progress**（[*bool*](https://docs.python.org/3/library/functions.html#bool "(in Python v3.12)")*，*可选） - 是否在 stderr 上显示进度条。默认值：True

+   **check_hash**（[*bool*](https://docs.python.org/3/library/functions.html#bool "(in Python v3.12)")*，*可选） - 如果为 True，则 URL 的文件名部分应遵循命名约定`filename-<sha256>.ext`，其中`<sha256>`是文件内容的 SHA256 哈希的前八位或更多位数字。哈希用于确保唯一名称并验证文件的内容。默认值：False

+   **file_name**（[*str*](https://docs.python.org/3/library/stdtypes.html#str "(in Python v3.12)")*，*可选） - 下载文件的名称。如果未设置，则将使用`url`中的文件名。

+   **weights_only**（[*bool*](https://docs.python.org/3/library/functions.html#bool "(in Python v3.12)")*，*可选） - 如果为 True，则仅加载权重，不加载复杂的 pickled 对象。建议用于不受信任的来源。有关更多详细信息，请参阅`load()`。

返回类型

[*Dict*](https://docs.python.org/3/library/typing.html#typing.Dict "(in Python v3.12)")[[str](https://docs.python.org/3/library/stdtypes.html#str "(in Python v3.12)"), [*Any*](https://docs.python.org/3/library/typing.html#typing.Any "(in Python v3.12)")]

示例

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