timm----深度学习预训练模型库使用
timm学习记录
·
1、安装与使用
首先安装两个库
pip install timm
pip install huggingface_hub
2、创建预训练模型
代码很简单就一句
model = timm.create_model('resnet18', pretrained=True, num_classes=4)
model = timm.create_model("hf_hub:timm/eca_nfnet_l0")
需要什么模型都可以去他们的官网上查看Models - Hugging Face
| [](http://www.relxdingyilang.cn/) | [点击访问博客查看更多内容](http://www.relxdingyilang.cn/) |
|------------------------------------------|-- |
更多推荐
所有评论(0)