移至 <cite>torch.hub</cite> 。
torch.utils.model_zoo.load_url(url, model_dir=None, map_location=None, progress=True, check_hash=False)¶
将 Torch 序列化对象加载到给定的 URL。
如果下载的文件是 zip 文件,它将被自动解压缩。
如果 <cite>model_dir</cite> 中已经存在该对象,则将其反序列化并返回。 <cite>model_dir</cite> 的默认值为$TORCH_HOME/checkpoints
,其中环境变量$TORCH_HOME
的默认值为$XDG_CACHE_HOME/torch
。 $XDG_CACHE_HOME
遵循 Linux 文件系统布局的 X 设计组规范,如果未设置,则默认值为~/.cache
。
参数
url (字符串)–要下载的对象的 URL
model_dir (字符串 , 可选)–保存对象的目录
map_location (可选)–指定如何重新映射存储位置的函数或命令(请参见 torch.load)
进度 (bool , 可选)–是否显示 stderr 进度条。 默认值:True
check_hash (bool , 可选)–如果为 True,则 URL 的文件名部分应遵循命名约定filename-<sha256>.ext
,其中[ <sha256>
是文件内容的 SHA256 哈希值的前 8 位或更多位。 哈希用于确保唯一的名称并验证文件的内容。 默认值:False
例
>>> state_dict = torch.hub.load_state_dict_from_url('https://s3.amazonaws.com/pytorch/models/resnet18-5c106cde.pth')
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。