• 文档 >
  • torch.utils.model_zoo
快捷方式

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

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

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

  • map_location (optional) – 指定如何重新映射存储位置的函数或字典(参见 torch.load)

  • progress (bool, optional) – 是否在 stderr 上显示进度条。默认值:True

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

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

  • weights_only (bool, optional) – 如果为 True,将只加载权重,不会加载复杂的腌制对象。建议用于不受信任的来源。有关详细信息,请参见 load()

返回类型

Dict[str, Any]

示例

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源