pip install timm
import timm import torch
## Model List list_models = timm.list_models(pretrained=True)
## Create Model model = timm.create_model('resnet50', pretrained=True) model.eval()
https://github.com/huggingface/pytorch-image-models