当前位置: 首页 > 工具软件 > model-zoo > 使用案例 >

torch.utils.model_zoo.load_url的用法

吴子昂
2023-12-01

平时加载预训练模型的时候,使用pytorch自带的model_zoo方法加载url地址中的模型,具体的api如下:

torch.utils.model_zoo.load_url(url, model_dir=None)

作用:在给定URL上加载torch序列化对象。
具体来讲就是通过提供的.pth文件的url地址来下载指定的.pth文件。
参数:
url (string) - 要下载对象的URL
model_dir (string, optional) - 保存对象的目录
如果对象已经存在于model_dir中,则将被反序列化并返回。【也就是后面所说的——已经下载好模型的情况】
详细说明:
如果对象已经存在于 model_dir 中,则将被反序列化并返回。URL的文件名部分应遵循命名约定filename-.ext,其中是文件内容的SHA256哈希的前八位或更多位数字。哈希用于确保唯一的名称并验证文件的内容。model_dir 的默认值为 T O R C H H O M E / m o d e l s ,其中 TORCH_HOME/models,其中 TORCHHOME/models,其中TORCH_HOME默认为~/.torch。可以使用$TORCH_MODEL_ZOO环境变量来覆盖默认目录。

 类似资料: