facebookresearch / swav

PyTorch implementation of SwAV https//arxiv.org/abs/2006.09882
Other
2.01k stars 280 forks source link

ImageNet pretraining size #116

Open slahrichi opened 1 year ago

slahrichi commented 1 year ago

Hi! Thank you for this awesome paper. My team and I have been using SwAV heavily in our work. I am trying to train your model from scratch on ImageNet and was wondering if you use the ILSVRC 1.2M subset, or if you use the entire dataset (~14M images). Many thanks!