timm从本地加载预训练模型

想要从timm加载本地预训练模型,首先是参考timm.create_model()从本地加载pretrained模型

model = timm.create_model('modelxxx', pretrained=True,  xxx)

改为

pretrained_cfg = timm.models.create_model("modelxxx").default_cfg
pretrained_cfg['file'] = 'path/to/checkpoint'
model = timm.models.create_model("modelxxx", pretrained=True, xxx, pretrained_cfg=pretrained_cfg))  

但是遇到错误

AssertionError: pretrained_cfg should not be set when sourcing model from Hugging Face Hub.

后来参考LocalEntryNotFoundError when loading downloaded pretrained model using timm.create_model (side load offline weights, e.g. on Kaggle) #1826
成功将从本地加载预训练模型

timm.create_model(
  'modelxxx',
  pretrained=True,
  pretrained_cfg_overlay=dict(file='path/to/checkpoint'),
)
相关推荐
糖豆豆今天也要努力鸭1 分钟前
torch.__version__的torch版本和conda list的torch版本不一致
linux·pytorch·python·深度学习·conda·torch
何大春17 分钟前
【弱监督语义分割】Self-supervised Image-specific Prototype Exploration for WSSS 论文阅读
论文阅读·人工智能·python·深度学习·论文笔记·原型模式
在下不上天25 分钟前
Flume日志采集系统的部署,实现flume负载均衡,flume故障恢复
大数据·开发语言·python
SEVEN-YEARS29 分钟前
深入理解TensorFlow中的形状处理函数
人工智能·python·tensorflow
EterNity_TiMe_33 分钟前
【论文复现】(CLIP)文本也能和图像配对
python·学习·算法·性能优化·数据分析·clip
Suyuoa1 小时前
附录2-pytorch yolov5目标检测
python·深度学习·yolo
好看资源平台2 小时前
网络爬虫——综合实战项目:多平台房源信息采集与分析系统
爬虫·python
进击的六角龙2 小时前
深入浅出:使用Python调用API实现智能天气预报
开发语言·python
檀越剑指大厂2 小时前
【Python系列】浅析 Python 中的字典更新与应用场景
开发语言·python
湫ccc2 小时前
Python简介以及解释器安装(保姆级教学)
开发语言·python