参考：https://rwightman.github.io/pytorch-image-models/
该库包含了很多种类的涉及图像模型，包括ViT。
安装pip install timm #或者 pip install git+https://github.com/rwightman/pytorch-image-models.git读取模型 查看有哪些可用模型
查看有哪些 pre-trained模型可用：
import timm from pprint import pprint model_names = timm.list_models(pretrained=True) pprint(model_names)
out：
并且可以找到我们需要的ViT：
比如说我要找到包含 “vit” 字符的模型：
model_names = timm.list_models('*vit*')
pprint(model_names)
读取模型
举例：
import timm
m = timm.create_model('mobilenetv3_large_100', pretrained=True)
m.eval()
结尾
这里有一个不错的教程，有时间可以看看：
https://towardsdatascience.com/getting-started-with-pytorch-image-models-timm-a-practitioners-guide-4e77b4bf9055
