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