当前位置: 首页 > news >正文

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'),
)

http://www.mrgr.cn/news/21118.html

相关文章:

  • Jmeter模拟用户登录时获取token如何跨线程使用?
  • leetcode:516 最长回文字序列 动态规划
  • 夫妻离婚对债务的约定是否对第三人有效?
  • [Android] [SnapdragonCamera] 单摄(横屏)阶段总结
  • GitHub每日最火火火项目(9.6)
  • Git 撤回commit
  • 主流的3D模型格式有什么区别?
  • <数据集>二维码识别数据集<目标检测>
  • 罗盘时钟屏保你见过吗?非常有特色的电脑时钟屏保
  • docker原理以及使用
  • 利用KMeans重新计算自己数据集的anchor
  • C/C++的自由落体运动
  • 软件测试学习笔记丨Pytest的使用
  • Qt/C++开源项目 TCP客户端调试助手(源码分享+发布链接下载)
  • GIT:git add命令指定文件夹
  • Linux多线程——互斥锁的封装与生产消费模型的实现
  • MySQL你必须知道的事
  • (感知机-Perceptron)—有监督学习方法、非概率模型、判别模型、线性模型、参数化模型、批量学习、核方法
  • 解决安装mpi4py失败的问题
  • python恶搞无限弹窗脚本