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

question #16

Open Michael-Chao opened 1 year ago

Michael-Chao commented 1 year ago

Do I need to import a pre trained model after joining the network for my model?