facebookresearch / moco

PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722
MIT License
4.83k stars 794 forks source link

mocov2 performance curve for different K #28

Closed srikar2097 closed 4 years ago

srikar2097 commented 4 years ago

Hi @KaimingHe , We had fun reading your paper and thank you for sharing your work.

Fig. 3 in mocov1 paper compares various contrastive learning mechanisms accuracy for varying K. We could not find this plot for mocov2. Could you please share this for mocov2? Basically acc for those 6 points K=(256, ... 65536)

thanks again, Srikar

endernewton commented 4 years ago

We did not show the curve of changing K in the mocov2 report, but according to our experiments mocov2 is quite resilient to the change of K (down to some limit). You could also run the experiments yourself, for 256, 1024, etc, should not take long.