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'),
)
相关推荐
nbsaas-boot4 小时前
Java 正则表达式白皮书:语法详解、工程实践与常用表达式库
开发语言·python·mysql
仗剑_走天涯4 小时前
基于pytorch.nn模块实现线性模型
人工智能·pytorch·python·深度学习
chao_7894 小时前
二分查找篇——搜索旋转排序数组【LeetCode】两次二分查找
开发语言·数据结构·python·算法·leetcode
chao_7899 小时前
二分查找篇——搜索旋转排序数组【LeetCode】一次二分查找
数据结构·python·算法·leetcode·二分查找
烛阴9 小时前
Python装饰器解除:如何让被装饰的函数重获自由?
前端·python
noravinsc10 小时前
django 一个表中包括id和parentid,如何通过parentid找到全部父爷id
python·django·sqlite
ajassi200010 小时前
开源 python 应用 开发(三)python语法介绍
linux·python·开源·自动化
沉默媛10 小时前
如何安装python以及jupyter notebook
开发语言·python·jupyter
Deng94520131411 小时前
基于Python的旅游数据可视化应用
python·numpy·pandas·旅游·数据可视化技术
2401_8786247911 小时前
pytorch 自动微分
人工智能·pytorch·python·机器学习