当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python PyTorch load_url用法及代码示例


本文简要介绍python语言中 torch.utils.model_zoo.load_url 的用法。

用法:

torch.utils.model_zoo.load_url(url, model_dir=None, map_location=None, progress=True, check_hash=False, file_name=None)

参数

  • url(string) -要下载的对象的 URL

  • model_dir(string,可选的) -保存对象的目录

  • map_location(可选的) -指定如何重新映射存储位置的函数或字典(请参阅 torch.load)

  • progress(bool,可选的) -是否向 stderr 显示进度条。默认值:真

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

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

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

如果下载的文件是 zip 文件,则会自动解压缩。

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

示例

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

相关用法


注:本文由纯净天空筛选整理自pytorch.org大神的英文原创作品 torch.utils.model_zoo.load_url。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。