Question about the test-time image augmentations

Thanks for your wonderful work!

Could you please provide the codes or implementation instructions for test-time image augmentations used in Figure 2, such as Diffusion-Weak (‘W’) and Diffusion-Strong (’S’), mixtures: strategies adopted in MOCO, BYOL and SimCLR?