facebookresearch/swav

Why using the hyperparameter "freeze_prototypes_niters"?

xuChenSJTU opened this issue · 1 comments

xuChenSJTU commented

Hi,

Thanks for sharing the codes. I have some questions and could help me figure them out?

  1. Why using the hyperparameter "freeze_prototypes_niters", is it important for the model performance?
  2. Why setting it as "313"? A weird number?

Thanks.

mathildecaron31 commented
  1. not important for model performance, but important for model stability in some cases
  2. 313 is the number of iterations for one epoch on ImageNet with batch size of 4096. It means that we freeze the prototypes for one epoch.