search

Sara-Ahmed / SiT

Self-supervised vIsion Transformer (SiT)
324 stars 49 forks source link

checkpoints #5

Closed Booooooooooo closed 3 years ago

Booooooooooo commented 3 years ago

Thank you for your excellent work! I am interested in fine-tuning SiT for other tasks. Will you consider providing the pre-trained models?

Sara-Ahmed commented 3 years ago

Thanks a lot. Of course. We trained SiT on CIFAR10, CIFAR100, STL10, TinyImageNet, and 10% of ImageNet. We did not manage to train the network on full ImageNet as it is computationally expensive.

Sara-Ahmed commented 3 years ago

Added!