Did you train the model with some datasets like ImageNet-1k?

chinhsuanwu / mobilevit-pytorch

A PyTorch implementation of "MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformer"

https://arxiv.org/abs/2110.02178

MIT License

501 stars 70 forks source link

Open tucan9389 opened 3 years ago

tucan9389 commented 3 years ago

If you have, could you share the result of the experiments?

Anyway, thank you for your sharing the clear implementation!

chinhsuanwu commented 3 years ago

Hi @tucan9389

I haven't trained on ImageNet, please refer to this, but I might try to.

tucan9389 commented 3 years ago

i'll be waiting for your result 😁 if i make some result later, i'll share it also. thank you.

chinhsuanwu commented 3 years ago

Sure, welcome.

adityagupta26 commented 3 years ago

can you guide me how to train this model , I want to give it a try