python使用timm创建模型时报错:connection error, and we cannot find the requested files in the cached path.

解决步骤:这里借鉴了其他博主的方法,尝试了好多方法,这里总结一下我成功的方法
这里我想用的是resnet18这个模型,以这个为例,用其他模型只需要把模型名字输进去替换掉我这里的resnet18

import timm
print(timm.models.create_model('resnet18').default_cfg)

输入这个代码之后运行,会出现以下链接

{'url': 'https://github.com/huggingface/pytorch-image-models/releases/download/v0.1-rsb-weights/resnet18_a1_0-d63eafa0.pth', 'hf_hub_id': 'timm/resnet18.a1_in1k', 'architecture': 'resnet18', 'tag': 'a1_in1k', 'custom_load': False, 'input_size': (3, 224, 224), 'test_input_size': (3, 288, 288), 'fixed_input_size': False, 'interpolation': 'bicubic', 'crop_pct': 0.95, 'test_crop_pct': 1.0, 'crop_mode': 'center', 'mean': (0.485, 0.456, 0.406), 'std': (0.229, 0.224, 0.225), 'num_classes': 1000, 'pool_size': (7, 7), 'first_conv': 'conv1', 'classifier': 'fc', 'origin_url': 'https://github.com/huggingface/pytorch-image-models', 'paper_ids': 'arXiv:2110.00476'}

这里边的 ‘hf_hub_id’: ‘timm/resnet18.a1_in1k’ 这个就是我们需要的权重地址
我标灰的这个加上前缀 https://huggingface.co/
也就是变成了 https://huggingface.co/timm/resnet18.a1_in1k这个样子

'timm/resnet18.a1_in1k在这里插入图片描述
翻墙之后去谷歌:粘贴这个地址,打开huggingface的官网(只有翻出去才能打开应该)
在这里插入图片描述
打开这个里边

在这里插入图片描述
下载对应的bin文件
在这里插入图片描述
下载之后的bin文件放在c盘
C:\Users\14717(这里是我的,放在你的用户名下).cache\huggingface\hub\models–timm–resnet18.a1_in1k
在这里插入图片描述
然后就可以创建你想创建的网络
**代码如下:*里边的地址file对应的拷贝你存放刚刚bin文件的位置,但是要加上这个bin文件
C:\Users\14717.cache\huggingface\hub\models–timm–resnet18.a1_in1k(这个是拷贝的存放地址
但是你要加上这个文件的文件名,变成这样
C:\Users\14717.cache\huggingface\hub\models–timm–resnet18.a1_in1k*pytorch_model.bin

防止读不出来要加上r或者用\两个反斜杠
这个时候在创建模型就可以了

net= timm.create_model("resnet18", pretrained=True, pretrained_cfg_overlay=dict(file=r'C:\Users\14717\.cache\huggingface\hub\models--timm--resnet18.a1_in1k\pytorch_model.bin'))
net=net.cuda(0)

我这里用的名字是net,大家可以自己更换名字如model…等其他的

  • 8
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值