Will you provide the pretrained model of CaiT-XXS-24 or CaiT-XXS-36?

facebookresearch / deit

Official DeiT repository

Apache License 2.0

4.07k stars 556 forks source link

Will you provide the pretrained model of CaiT-XXS-24 or CaiT-XXS-36? #87

Closed JiaDingCN closed 3 years ago

JiaDingCN commented 3 years ago

Hi, I have read your CaiT paper and the performance in this paper is very good. I want to learn more about that and the CaiT-XXS model seems suitable to my computer's GPU. But the cait_models.py file in this repository seems do not contain the pretrained model of CaiT-XXS. Do you plan to provide it? Thanks.

TouvronHugo commented 3 years ago

Hi @JiaDingCN , Thanks for your message, I have just added the XXS models with pre-trained weight for resolution 224x224 and 384x384. Best, Hugo

JiaDingCN commented 3 years ago

It really solves my problem, thank you!