Pytorch的timm库(Timm library)(torchvision.models 的扩展版)(CV为主)

Timm library

这个库可以算是torchvision.models 的扩展版，包含的模型以CV的分类任务为主，包含模型的准确率要高不少

号称打kaggle必备库之一

DeiT就是直接用的Timm中的ViT code

像我们在自己跑一些简单小实验的时候, 调用一些模型，直接一句话就行了，不用再写什么resnet的代码了

import timm
model = timm.create_model('efficientnet_b0', in_chans=1, num_classes=10).cuda()

查看timm可以直接用的模型有哪些

print(timm.list_models())
#或
print(timm.list_models('resnet*'))

查看模型权重的下载路径

如果网络不通无法下载的话

print(timm.models.create_model('resnet18').default_cfg)

加载下载到本地的模型

resnet = timm.create_model('resnet18', 
                          pretrained=True,
                          pretrained_cfg_overlay=dict(file='resnet18_a1_0-d63eafa0.pth'))

一个好用的预训练库timm(PyTorch) - 知乎