search

sthalles / SimCLR

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations
https://sthalles.github.io/simple-self-supervised-learning/
MIT License
2.19k stars 458 forks source link

About learning rate schedule #43

Closed theneotopia closed 2 years ago

theneotopia commented 2 years ago

Hi, Thalles. https://github.com/sthalles/SimCLR/blob/1848fc934ad844ae630e6c452300433fe99acfd9/run.py#L79 Shouldn't T_max=args.epochs instead of T_max=len(train_loader) since learning rate schedule happens at every epoch? Thanks.