search

apple / ml-cvnets

CVNets: A library for training computer vision networks
https://apple.github.io/ml-cvnets
Other
1.76k stars 225 forks source link

MobileViT (version 1) pre-training dataset #56

Closed YHYeooooong closed 1 year ago

YHYeooooong commented 1 year ago

Hello! thanks for sharing the great work!

I have a small question about the pre-training dataset. Did all MobileViT (version 1) models pre-trained on ImageNet-1k? I guess they did, but I want to make sure. Thanks!

mobilevit

sayannath commented 1 year ago

Yes they are pre-trained weights of the ImageNet-1k dataset.

YHYeooooong commented 1 year ago

Thanks!

JUNSUNGKIM99 commented 1 year ago

Hello! Thanks for your effort!

I also have a question about dataset. In my situation, i'm going to test the cvnets briefly. There was a problem setting the dataset in the config file. Do I actually have to download imagenet 1k dataset?

Thanks!

farzadab commented 1 year ago

Yes, you’ll have to download the dataset and then set train_root and val_root to point to where the dataset is located.

On Wed, Feb 22, 2023 at 12:48 AM JunSung Kim @.***> wrote:

Hello! Thanks for your effort!

I also have a question about dataset. In my situation, i'm going to test the cvnets briefly. There was a problem setting the dataset in the config file. Do I actually have to download imagenet 1k dataset?

Thanks!

— Reply to this email directly, view it on GitHub https://github.com/apple/ml-cvnets/issues/56#issuecomment-1439636915, or unsubscribe https://github.com/notifications/unsubscribe-auth/ABGT6OUBOI73ZJCKZPGJP33WYXHEBANCNFSM6AAAAAARGJPDRY . You are receiving this because you are subscribed to this thread.Message ID: @.***>