9、TORCH.UTILS.MODEL_ZOO

本文主要是介绍9、TORCH.UTILS.MODEL_ZOO，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

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

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

如果下载的文件是 zip 文件，它会被自动解压。

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

Parameters

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

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

• map_location (optional) – a function or a dict specifying how to remap storage locations (see torch.load)

• progress (bool, optional) – 是否向 stderr 显示进度条。 默认值：真

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

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

例子：

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

这篇关于9、TORCH.UTILS.MODEL_ZOO的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---