Question about how to pre-training the simCLR model with CIFAR-10?

google-research / simclr

SimCLRv2 - Big Self-Supervised Models are Strong Semi-Supervised Learners

https://arxiv.org/abs/2006.10029

Apache License 2.0

4.07k stars 622 forks source link

Question about how to pre-training the simCLR model with CIFAR-10? #126

Open r000ot opened 3 years ago

r000ot commented 3 years ago

Hi Ting,

Thank you for sharing the code! I want to inquire some question about pre-training SimCLR on CIFAR-10 with one GPU. I used the command of pre-training SimCLR on CIFAR-10, it ran the whole night with no output. For your convenience, could you please give me some guidance of how to solve this problem? Thank you for your time and kind help!

Best, Yangru

chentingpc commented 3 years ago

Are you using TF2 implementation? It should print logging information as it runs.

r000ot commented 3 years ago

Thank you so much for your kind help! I run the code successfully now.