python_timm手动下载模型

使用timm模块中，create_model初始化模型，国内无法连接Hugging Face网站，通常会遇到下载失败的错误。

from timm import create_model
import timm

model_list=timm.list_models()
print(model_list)

model=create_model(
    'resnet18',
    pretrained=False,
    features_only=True
)

print(model.default_cfg)


model1=create_model(
    'resnet18',
    pretrained=True,
    pretrained_cfg_overlay=dict(file='./resnet18_a1_0-d63eafa0.pth')
)

1.查看现有的模型名称

  model_list=timm.list_models()

  print(model_list)

2.查看模型的链接地址，先将pretrained设置为False，初始化模型之后，打印输出模型的链接地址。

print(model.default_cfg)

用科学上网工具，根据该链接地址下载模型。

3.将下载的模型拷贝到工程目录下，在pretrained_cfg_overlay的参数中输入该模型的路径，即可成功初始化该模型。