SimCLR icon indicating copy to clipboard operation
SimCLR copied to clipboard

Should assert n_views == 2?

Open LinglanZhao opened this issue 3 years ago • 1 comments

Thanks for your excellent implementation! I'd like to confirm that N_VIEW == 2 as in the paper and the default args in the code. If N_VIEW > 2, with logits.shape = (N_VIEW x N, N_VIEW x N - 1) https://github.com/sthalles/SimCLR/blob/1848fc934ad844ae630e6c452300433fe99acfd9/simclr.py#L51, N_VIEW x N - 1 contains at least one more positive pairs (except the one indexed with 0) which will be treated as negative pairs. @sthalles @alessiamarcolini @butyuhao

LinglanZhao avatar Apr 17 '21 14:04 LinglanZhao

Have you solved the problem? I meet the same question. Thanks.

here101 avatar Dec 30 '21 11:12 here101