search

THU-MIG / RepViT

RepViT: Revisiting Mobile CNN From ViT Perspective [CVPR 2024] and RepViT-SAM: Towards Real-Time Segmenting Anything
https://arxiv.org/abs/2307.09283
Apache License 2.0
730 stars 55 forks source link

Model classifier weights for CIFAR100 #3

Closed moliqingwa closed 1 year ago

moliqingwa commented 1 year ago

Hi, will you release the trained classifier weights for all other datasets instead of IMNET ? I checked the released repvit_m3 model weights it seems the classifier is for 1000 classes. Thanks

jameslahm commented 1 year ago

Hi, thanks, we have not trained repvit on CIFAR100 yet. I suggest that you could also use the weights pretrained on IMNET to finetune repvit on CIFAR100 as well. And I am sorry that I don't have bandwidth for training repvit on CIFAR100 at the moment.

moliqingwa commented 1 year ago

Got it, thanks