平时加载预训练模型的时候,使用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环境变量来覆盖默认目录。